Merge pull request #18 from ubc-systopia/fix-sklearn-import

Fix import statement to work with latest sklearn
ubc-systopia · Jan 21, 2025 · 48d9cc8 · 48d9cc8
2 parents 160af59 + 653e3f7
commit 48d9cc8
Showing 1 changed file with 2 additions and 1 deletion.
diff --git a/src/gosdt/_threshold_guessing.py b/src/gosdt/_threshold_guessing.py
@@ -1,7 +1,8 @@
 # External imports
 import numpy as np
 import pandas as pd
-from sklearn.base import BaseEstimator, TransformerMixin, check_X_y
+from sklearn.base import BaseEstimator, TransformerMixin
+from sklearn.utils import check_X_y
 from sklearn.ensemble import GradientBoostingClassifier
 from sklearn.utils.validation import check_array, check_is_fitted