From bf22b9b11489492280d11266739b1ce24b506719 Mon Sep 17 00:00:00 2001
From: wenfeiy-db <wenfei.yan@databricks.com>
Date: Thu, 11 Jan 2024 11:10:34 -0800
Subject: [PATCH] replace

---
 .../databricks/automl_runtime/sklearn/datetime_imputer.py | 2 +-
 runtime/environment.txt                                   | 8 ++++----
 2 files changed, 5 insertions(+), 5 deletions(-)

diff --git a/runtime/databricks/automl_runtime/sklearn/datetime_imputer.py b/runtime/databricks/automl_runtime/sklearn/datetime_imputer.py
index c897f428..32880cb7 100644
--- 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
index a52cc682..44a96bc1 100644
--- 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