060a71ef947f9d9e87e94dd8320ef78c83c946cb,tests/xgboost/test_xgboost_converters.py,TestXGBoostModels,test_xgb_classifier_multi,#TestXGBoostModels#,41

Before Change

    @unittest.skipIf(sys.version_info[0] == 2, reason="xgboost converter not tested on python 2")
    def test_xgb_classifier_multi(self):
        iris = load_iris()
        X = iris.data[:, :2]
        y = iris.target

        xgb = XGBClassifier()
        xgb.fit(X, y)
        conv_model = convert_xgboost(xgb, initial_types=[("input", FloatTensorType(shape=[1, "None"]))])
        self.assertTrue(conv_model is not None)
        dump_multiple_classification(xgb, allow_failure="StrictVersion(onnx.__version__) < StrictVersion("1.3.0")")

After Change

    @unittest.skipIf(sys.version_info[0] == 2,
                     reason="xgboost converter not tested on python 2")
    def test_xgb_classifier_multi(self):
        xgb, x_test = _fit_classification_model(XGBClassifier(), 3)
        conv_model = convert_xgboost(
            xgb, initial_types=[("input", FloatTensorType(shape=[1, "None"]))])
        self.assertTrue(conv_model is not None)
        dump_data_and_model(
            x_test,
            xgb,
            conv_model,
            basename="SklearnXGBClassifierMulti",
            allow_failure="StrictVersion("
            "onnx.__version__)"
            "< StrictVersion("1.3.0")",
        )

    @unittest.skipIf(sys.version_info[0] == 2,
                     reason="xgboost converter not tested on python 2")
    def test_xgb_classifier_multi_reglog(self):

Instances