Bug fix

Fix calling convert_sklearn
sdpython · Aug 2, 2019 · 9ea1a0f · 9ea1a0f
1 parent a49d3d1
commit 9ea1a0f
Showing 1 changed file with 1 addition and 2 deletions.
diff --git a/mlprodict/onnxrt/validate_helper.py b/mlprodict/onnxrt/validate_helper.py
@@ -207,8 +207,7 @@ def to_onnx(model, X=None, name=None, initial_types=None,
     else:
         old_values = None
     res = convert_sklearn(model, initial_types=initial_types, name=name,
-                          target_opset=target_opset, options=options,
-                          dtype=new_dtype)
+                          target_opset=target_opset, options=options)
     if old_values is not None:
         register_rewritten_operators(old_values)
     return res