Simplify import path to add_onnx_graph (#1024)

Signed-off-by: Xavier Dupre <xadupre@microsoft.com>
onnx · Sep 5, 2023 · 9be63df · 9be63df
1 parent 052f51d
commit 9be63df
Show file tree

Hide file tree

Showing 3 changed files with 3 additions and 2 deletions.
diff --git a/docs/api_summary.rst b/docs/api_summary.rst
@@ -66,7 +66,7 @@ lists all available converters.
 Helpers for new converters
 ==========================
 
-.. autofunction:: skl2onnx.helpers.integration.add_onnx_graph
+.. autofunction:: skl2onnx.helpers.add_onnx_graph
 
 Manipulate ONNX graphs
 ======================

diff --git a/skl2onnx/helpers/__init__.py b/skl2onnx/helpers/__init__.py
@@ -3,3 +3,4 @@
 
 from .investigate import collect_intermediate_steps, compare_objects  # noqa
 from .investigate import enumerate_pipeline_models  # noqa
+from .integration import add_onnx_graph  # noqa
diff --git a/tests/test_other_converter_library_pipelines.py b/tests/test_other_converter_library_pipelines.py
@@ -18,7 +18,7 @@
 from skl2onnx.operator_converters.linear_classifier import (
     convert_sklearn_linear_classifier,
 )
-from skl2onnx.helpers.integration import add_onnx_graph
+from skl2onnx.helpers import add_onnx_graph
 from test_utils import dump_data_and_model, TARGET_OPSET