revert code suggestion

aimclub · Apr 4, 2024 · 6851860 · 6851860
1 parent c298dc0
commit 6851860
Showing 1 changed file with 7 additions and 6 deletions.
diff --git a/fedot/api/main.py b/fedot/api/main.py
@@ -442,12 +442,13 @@ def get_metrics(self,
                                            data_type=self.train_data.data_type)
             else:
                 self.test_data.target = target[:len(self.prediction.predict)]
-        elif not len(self.test_data.target):
-            self.test_data = self.data_processor.define_data(
-                target=self.train_data.target[:len(self.test_data.features)],
-                features=self.test_data.features,
-                is_predict=True,
-            )
+        else:
+            if not len(self.test_data.target):
+                self.test_data = self.data_processor.define_data(
+                    target=self.train_data.target[:len(self.test_data.features)],
+                    features=self.test_data.features,
+                    is_predict=True,
+                )
 
         metrics = ensure_wrapped_in_sequence(metric_names) if metric_names else self.metrics
         metric_names = [str(metric) for metric in metrics]