Enable support for scikit-learn 1.5

Signed-off-by: Keith Battocchi <kebatt@microsoft.com>

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,

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",