replace

wenfeiy-db · Jan 11, 2024 · bf22b9b · bf22b9b
1 parent f2437fc
commit bf22b9b
Show file tree

Hide file tree

Showing 2 changed files with 5 additions and 5 deletions.
diff --git a/runtime/databricks/automl_runtime/sklearn/datetime_imputer.py b/runtime/databricks/automl_runtime/sklearn/datetime_imputer.py
@@ -46,7 +46,7 @@ def fit(self, X, y=None):
         y: not used
         """
         self.fill_values = {}
-        for col_name, col_value in X.iteritems():
+        for col_name, col_value in X.items():
             col_value = pd.to_datetime(col_value, errors="coerce")
             if self.strategy == 'mean':
                 self.fill_values[col_name] = col_value.mean(skipna=True)

diff --git a/runtime/environment.txt b/runtime/environment.txt
@@ -4,11 +4,11 @@
 category_encoders==2.6.0
 holidays==0.28
 hyperopt==0.2.7
-mlflow==2.0.1
+mlflow==2.9.2
 numpy==1.23.5
-pandas==1.5.3
+pandas==2.0.3
 pmdarima==2.0.3
 prophet==1.1.4
-pyarrow==8.0.0
-scikit-learn==1.1.1
+pyarrow==14.0.1
+scikit-learn==1.3.0
 wrapt==1.14.1