mixed

wenfeiy-db · Jan 12, 2024 · c9ab5a0 · c9ab5a0
1 parent 857af6d
commit c9ab5a0
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/runtime/tests/automl_runtime/sklearn/datetime_imputer_test.py b/runtime/tests/automl_runtime/sklearn/datetime_imputer_test.py
@@ -29,10 +29,10 @@ def get_test_df(self, fill_value_1=None, fill_value_2=None):
     def test_imputer_mean(self):
         assert_frame_equal(
                 DatetimeImputer().fit_transform(self.get_test_df()),
-                self.get_test_df("2021-01-04 14:24", "2022-01-01 00:04:36").apply(pd.to_datetime))
+                self.get_test_df("2021-01-04 14:24", "2022-01-01 00:04:36").apply(lambda x: pd.to_datetime(x, format='mixed')))
         assert_frame_equal(
                 DatetimeImputer(strategy="mean").fit_transform(self.get_test_df()),
-                self.get_test_df("2021-01-04 14:24", "2022-01-01 00:04:36").apply(pd.to_datetime))
+                self.get_test_df("2021-01-04 14:24", "2022-01-01 00:04:36").apply(lambda x: pd.to_datetime(x, format='mixed')))
 
     def test_imputer_median(self):
         assert_frame_equal(