Change import

Quantco · Jul 15, 2024 · 0ce4c73 · 0ce4c73
1 parent caeea27
commit 0ce4c73
Show file tree

Hide file tree

Showing 4 changed files with 4 additions and 4 deletions.
diff --git a/tests/test_drlearner.py b/tests/test_drlearner.py
@@ -7,9 +7,9 @@
 import onnxruntime as rt
 import pytest
 from lightgbm import LGBMClassifier, LGBMRegressor
+from onnxconverter_common.data_types import FloatTensorType
 from onnxmltools import convert_lightgbm, convert_xgboost
 from skl2onnx import convert_sklearn
-from skl2onnx.common.data_types import FloatTensorType
 from sklearn.linear_model import LinearRegression, LogisticRegression
 from sklearn.neighbors import RadiusNeighborsRegressor
 from xgboost import XGBRegressor

diff --git a/tests/test_rlearner.py b/tests/test_rlearner.py
@@ -8,9 +8,9 @@
 import pandas as pd
 import pytest
 from lightgbm import LGBMClassifier, LGBMRegressor
+from onnxconverter_common.data_types import FloatTensorType
 from onnxmltools import convert_lightgbm, convert_xgboost
 from skl2onnx import convert_sklearn
-from skl2onnx.common.data_types import FloatTensorType
 from sklearn.linear_model import LinearRegression, LogisticRegression
 from xgboost import XGBRegressor
 

diff --git a/tests/test_tlearner.py b/tests/test_tlearner.py
@@ -8,8 +8,8 @@
 import onnxruntime as rt
 import pytest
 from lightgbm import LGBMClassifier, LGBMRegressor
+from onnxconverter_common.data_types import FloatTensorType
 from onnxmltools import convert_lightgbm, convert_xgboost
-from skl2onnx.common.data_types import FloatTensorType
 from skl2onnx.convert import convert_sklearn
 from sklearn.ensemble import (
     HistGradientBoostingClassifier,

diff --git a/tests/test_xlearner.py b/tests/test_xlearner.py
@@ -10,9 +10,9 @@
 import onnxruntime as rt
 import pytest
 from lightgbm import LGBMClassifier, LGBMRegressor
+from onnxconverter_common.data_types import FloatTensorType
 from onnxmltools import convert_lightgbm, convert_xgboost
 from skl2onnx import convert_sklearn
-from skl2onnx.common.data_types import FloatTensorType
 from sklearn.neighbors import RadiusNeighborsClassifier, RadiusNeighborsRegressor
 from xgboost import XGBClassifier, XGBRegressor