From a4a8659daabe8dd3fcf51e363f9d3d0cfe8986be Mon Sep 17 00:00:00 2001 From: Xavier Dupre Date: Thu, 31 Aug 2023 17:07:29 +0200 Subject: [PATCH] Simplify import path to add_onnx_graph Signed-off-by: Xavier Dupre --- docs/api_summary.rst | 2 +- skl2onnx/helpers/__init__.py | 1 + tests/test_other_converter_library_pipelines.py | 2 +- 3 files changed, 3 insertions(+), 2 deletions(-) diff --git a/docs/api_summary.rst b/docs/api_summary.rst index ccd29c3ef..24a0e2c89 100644 --- 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 index 79c95d55d..81f5f5e33 100644 --- 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 index bbc2d12fc..13b4fe26d 100644 --- 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