зеркало из https://github.com/py-why/EconML.git
Enable support for scikit-learn 1.5
Signed-off-by: Keith Battocchi <kebatt@microsoft.com>
This commit is contained in:
Родитель
589480471f
Коммит
784a8d45a5
|
@ -13,9 +13,14 @@ import numbers
|
|||
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,
|
||||
|
|
|
@ -30,7 +30,12 @@ from ...utilities import _RegressionWrapper, get_feature_names_or_default, inver
|
|||
# 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:
|
||||
|
|
|
@ -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",
|
||||
|
|
Загрузка…
Ссылка в новой задаче