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