typo fixed

ourownstory · Dec 7, 2022 · de8686d · de8686d
1 parent 20747fc
commit de8686d
Showing 1 changed file with 5 additions and 5 deletions.
diff --git a/neuralprophet/forecaster.py b/neuralprophet/forecaster.py
@@ -2670,12 +2670,12 @@ def _train(
                 )
                 self.model = time_net.TimeNet.load_from_checkpoint(checkpoint_callback.best_model_path)
 
-        if not metrics:
+        if not metrics_enabled:
             return None
-        else:
-            # Return metrics collected in logger as dataframe
-            metrics_df = pd.DataFrame(self.metrics_logger.history)
-            return metrics_df
+
+        # Return metrics collected in logger as dataframe
+        metrics_df = pd.DataFrame(self.metrics_logger.history)
+        return metrics_df
 
     def restore_trainer(self):
         """