Merge pull request #769 from a-szulc/master

add catch for autoML warning and a check for pandas future warning (fixes #754)
mljar · Sep 2, 2024 · 2b6aa16 · 2b6aa16
2 parents 3f65896 + fceff2b
commit 2b6aa16
Show file tree

Hide file tree

Showing 2 changed files with 10 additions and 1 deletion.
diff --git a/supervised/preprocessing/scale.py b/supervised/preprocessing/scale.py
@@ -45,6 +45,8 @@ def inverse_transform(self, X):
             if self.scale_method == self.SCALE_NORMAL:
                 X.loc[:, self.columns] = self.scale.inverse_transform(X[self.columns])
             elif self.scale_method == self.SCALE_LOG_AND_NORMAL:
+                X[self.columns] = X[self.columns].astype("float64")
+
                 X[self.columns] = self.scale.inverse_transform(X[self.columns])
                 X[self.columns] = np.exp(X[self.columns])
 

diff --git a/tests/tests_automl/test_targets.py b/tests/tests_automl/test_targets.py
@@ -317,7 +317,14 @@ def test_regression_missing_target(self):
             explain_level=0,
             start_random_models=1,
         )
-        automl.fit(X, y)
+
+        with pytest.warns(
+            match="There are samples with missing target values in the data which will be excluded for further analysis"
+        ) as record:
+            automl.fit(X, y)
+
+        self.assertEqual(len(record), 1)
+
         pred = automl.predict(X)
 
         self.assertIsInstance(pred, np.ndarray)