update test

Signed-off-by: Xavier Dupre <[email protected]>

tests/xgboost/test_xgboost_converters.py

@@ -288,7 +288,6 @@ def test_xgboost_booster_reg(self):
             n_classes=2, n_features=5, n_samples=100, random_state=42, n_informative=3
         )
         y = y.astype(np.float32) + 0.567
-        print(y)
         x_train, x_test, y_train, _ = train_test_split(
             x, y, test_size=0.5, random_state=42
         )
@@ -770,6 +769,8 @@ def test_xgb_classifier_13(self):
 
         initial_types = [("float_input", FloatTensorType([None, x_train.shape[1]]))]
         onnx_model = convert_xgboost(model, initial_types=initial_types)
+        # with open("debug.onnx", "wb") as f:
+        #     f.write(onnx_model.SerializeToString())
 
         expected = model.predict(x_test), model.predict_proba(x_test)
         sess = InferenceSession(onnx_model.SerializeToString())
@@ -779,4 +780,5 @@ def test_xgb_classifier_13(self):
 
 
 if __name__ == "__main__":
+    TestXGBoostModels().test_xgb_classifier_13()
     unittest.main(verbosity=2)