fix tests, replace mse with squared_error for tree based algorithms f…

…rom sklearn (#595)
mljar · Dec 30, 2022 · 66a3f52 · 66a3f52
1 parent 6850fb7
commit 66a3f52
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/supervised/preprocessing/label_encoder.py b/supervised/preprocessing/label_encoder.py
@@ -50,5 +50,5 @@ def to_json(self):
     def from_json(self, data_json):
         keys = np.array(list(data_json.keys()))
         if len(keys) == 2 and "False" in keys and "True" in keys:
-            keys = [False, True]
+            keys = np.array([False, True])
         self.lbl.classes_ = keys