diff --git a/quantile_forest/_quantile_forest.py b/quantile_forest/_quantile_forest.py index bc52a16..1c49eed 100755 --- a/quantile_forest/_quantile_forest.py +++ b/quantile_forest/_quantile_forest.py @@ -38,7 +38,11 @@ class calls the ``fit`` method of the ``ForestRegressor`` and creates a ) from sklearn.tree import DecisionTreeRegressor, ExtraTreeRegressor from sklearn.tree._tree import DTYPE -from sklearn.utils import parse_version + +try: + from sklearn.utils.fixes import parse_version +except ImportError: + from sklearn.utils import parse_version param_validation = True try: