Enable support for scikit-learn 1.5

Signed-off-by: Keith Battocchi <[email protected]>

Author: kbattocchi

econml/_ensemble/_ensemble.py

@@ -13,9 +13,14 @@
 import numpy as np
 from abc import ABCMeta, abstractmethod
 from sklearn.base import BaseEstimator, clone
-from sklearn.utils import _print_elapsed_time
 from sklearn.utils import check_random_state
 from joblib import effective_n_jobs
+from packaging.version import parse
+import sklearn
+if parse(sklearn.__version__) < parse("1.5"):
+    from sklearn.utils import _print_elapsed_time
+else:
+    from sklearn.utils._user_interface import _print_elapsed_time
 
 
 def _fit_single_estimator(estimator, X, y, sample_weight=None,

econml/solutions/causal_analysis/_causal_analysis.py

@@ -30,7 +30,12 @@
 # TODO: this utility is documented but internal; reimplement?
 from sklearn.utils import _safe_indexing
 # TODO: this utility is even less public...
-from sklearn.utils import _get_column_indices
+from packaging.version import parse
+import sklearn
+if parse(sklearn.__version__) < parse("1.5"):
+    from sklearn.utils import _get_column_indices
+else:
+    from sklearn.utils._indexing import _get_column_indices
 
 
 class _CausalInsightsConstants:

pyproject.toml

@@ -22,7 +22,7 @@ classifiers = [
 dependencies = [
     "numpy",
     "scipy > 1.4.0",
-    "scikit-learn >= 1.0, < 1.5",
+    "scikit-learn >= 1.0, < 1.6",
     "sparse",
     "joblib >= 0.13.0",
     "statsmodels >= 0.10",