Merge ifs

DubiousCactus · May 31, 2024 · 524cab4 · 524cab4
1 parent effa80c
commit 524cab4
Showing 1 changed file with 6 additions and 5 deletions.
diff --git a/src/base_trainer.py b/src/base_trainer.py
@@ -336,11 +336,12 @@ def _plot(self, epoch: int, train_losses: List[float], val_losses: List[float]):
             None
         """
         plt.clf()
-        if project_conf.LOG_SCALE_PLOT:
-            if any(loss_val <= 0 for loss_val in train_losses + val_losses):
-                raise ValueError(
-                    "Cannot plot on a log scale if there are non-positive losses."
-                )
+        if project_conf.LOG_SCALE_PLOT and any(
+            loss_val <= 0 for loss_val in train_losses + val_losses
+        ):
+            raise ValueError(
+                "Cannot plot on a log scale if there are non-positive losses."
+            )
         self._setup_plot(self._run_name, log_scale=project_conf.LOG_SCALE_PLOT)
         plt.plot(
             list(range(self._starting_epoch, epoch + 1)),