more fixing...

onnx · Apr 2, 2020 · 894d7c7 · 894d7c7
1 parent ee7fdad
commit 894d7c7
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/tests/xgboost/test_xgboost_pipeline.py b/tests/xgboost/test_xgboost_pipeline.py
@@ -133,7 +133,7 @@ def common_test_xgboost_10_skl(self, missing, replace=False):
         input_xgb = model.steps[0][-1].transform(test_df[:5]).astype(np.float32)
         if replace:
             input_xgb[input_xgb[:, :] == missing] = np.nan
-        onnx_last = convert_sklearn(model.steps[1][-1], input_xgb,
+        onnx_last = convert_sklearn(model.steps[1][-1],
                                     initial_types=[('input', FloatTensorType(shape=[None, input_xgb.shape[1]]))],
                                     target_opset=get_opset_number_from_onnx())
         session = rt.InferenceSession(onnx_last.SerializeToString())