black formatting

aidos-lab · Aug 16, 2024 · c8f76b5 · c8f76b5
1 parent 223fbc4
commit c8f76b5
Showing 1 changed file with 3 additions and 8 deletions.
diff --git a/code/models/base.py b/code/models/base.py
@@ -67,7 +67,7 @@ def general_step(self, batch, batch_idx, step: str):
             batch_size=batch_len,
             on_step=False,
             on_epoch=True,
-            sync_dist=True
+            sync_dist=True,
         )
         self.log_accuracies(x_hat, batch.y, batch_len, step)
         return loss
@@ -90,20 +90,15 @@ def log_accuracies(
         else:
             ValueError("Unknown step Literal")
 
-        accuracies = self.accuracies_fn(
-            acc_fun,
-            x_hat,
-            y,
-            step
-        )
+        accuracies = self.accuracies_fn(acc_fun, x_hat, y, step)
         for accuracy in accuracies:
             self.log(
                 **accuracy,
                 prog_bar=True,
                 on_step=False,
                 on_epoch=True,
                 batch_size=batch_len,
-                sync_dist=True
+                sync_dist=True,
             )
 
     def test_step(self, batch, batch_idx):