black eww

deepvk · Jan 30, 2024 · deada38 · deada38
1 parent d0af06e
commit deada38
Show file tree

Hide file tree

Showing 2 changed files with 4 additions and 4 deletions.
diff --git a/src/core/train.py b/src/core/train.py
@@ -341,7 +341,7 @@ def train_dim(args):
         )
 
         # Early stopping    -----------------------------------------------
-        if earl_stop:
+        if early_stop:
             logger.info(
                 "--> Early stopping. best_r_p {:0.2f} best_rmse {:0.2f}".format(
                     early_stop.best_r_p, early_stop.best_rmse
@@ -350,7 +350,7 @@ def train_dim(args):
             return
 
             # Training done --------------------------------------------------------
-    logger.info("--> Training done. best_r_p {:0.2f} best_rmse {:0.2f}".format(
-        early_stop.best_r_p, early_stop.best_rmse)
+    logger.info(
+        "--> Training done. best_r_p {:0.2f} best_rmse {:0.2f}".format(early_stop.best_r_p, early_stop.best_rmse)
     )
     return
diff --git a/src/utils/train_utils.py b/src/utils/train_utils.py
@@ -109,7 +109,7 @@ def get_loss(self, yb, yb_hat, idx):
         b = b[idx, :]
 
         yb_hat_map = (
-                b[:, 0] + b[:, 1] * yb_hat[:, 0] + b[:, 2] * yb_hat[:, 0] ** 2 + b[:, 3] * yb_hat[:, 0] ** 3
+            b[:, 0] + b[:, 1] * yb_hat[:, 0] + b[:, 2] * yb_hat[:, 0] ** 2 + b[:, 3] * yb_hat[:, 0] ** 3
         ).view(-1, 1)
 
         loss_bias = self._nan_mse(yb_hat_map, yb)