Needed some changes in astype to pass

PythonPredictions · Nov 3, 2023 · c6c3c56 · c6c3c56
1 parent 6130a4b
commit c6c3c56
Showing 1 changed file with 12 additions and 12 deletions.
diff --git a/tests/preprocessing/test_preprocessor.py b/tests/preprocessing/test_preprocessor.py
@@ -357,13 +357,13 @@ def test_mutable_train_data_fit_transform(self, mocker: MockerFixture):
                     },
                 ).astype(
                     {
-                        "a": np.float64(),
-                        "b": np.float64(),
-                        "d": np.float64(),
-                        "e": np.float64(),
-                        "category_1": pd.CategoricalDtype(),
-                        "category_2": pd.CategoricalDtype(),
-                        "category_3": pd.CategoricalDtype(),
+                        "a": "float64",
+                        "b": "float64",
+                        "d": "float64",
+                        "e": "float64",
+                        "category_1": "category",
+                        "category_2": "category",
+                        "category_3": "category",
                     }
                 ),
                 pd.DataFrame(
@@ -376,11 +376,11 @@ def test_mutable_train_data_fit_transform(self, mocker: MockerFixture):
                     }
                 ).astype(
                     {
-                        "a": np.float64(),
-                        "d": np.float64(),
-                        "e": np.float64(),
-                        "category_1": pd.CategoricalDtype(),
-                        "category_2": pd.CategoricalDtype(),
+                        "a": "float64",
+                        "d": "float64",
+                        "e": "float64",
+                        "category_1": "category",
+                        "category_2": "category",
                     }
                 ),
             ),