Update criterion in _interpreters.py

Author: 00helloworld

econml/cate_interpreter/_interpreters.py

@@ -4,6 +4,8 @@
 import abc
 import numbers
 import numpy as np
+from packaging import version
+import sklearn
 from sklearn.tree import DecisionTreeRegressor, DecisionTreeClassifier
 from sklearn.utils import check_array
 from ..policy import PolicyTree
@@ -149,7 +151,7 @@ def __init__(self, *,
         self.include_uncertainty = include_model_uncertainty
         self.uncertainty_level = uncertainty_level
         self.uncertainty_only_on_leaves = uncertainty_only_on_leaves
-        self.criterion = "mse"
+        self.criterion = "squared_error" if version.parse(sklearn.__version__) >= version.parse("1.0") else "mse"
         self.splitter = splitter
         self.max_depth = max_depth
         self.min_samples_split = min_samples_split