Format code with black

jameschapman19 · Sep 21, 2023 · ce2f082 · ce2f082
1 parent aab530c
commit ce2f082
Showing 1 changed file with 13 additions and 3 deletions.
diff --git a/cca_zoo/deep/_base.py b/cca_zoo/deep/_base.py
@@ -58,23 +58,33 @@ def training_step(self, batch: Dict[str, Any], batch_idx: int) -> torch.Tensor:
         loss = self.loss(batch["views"])
         for k, v in loss.items():
             # Use f-string instead of concatenation
-            self.log(f"train/{k}", v, prog_bar=True, on_epoch=True, batch_size=batch["views"][0].shape[0])
+            self.log(
+                f"train/{k}",
+                v,
+                prog_bar=True,
+                on_epoch=True,
+                batch_size=batch["views"][0].shape[0],
+            )
         return loss["objective"]
 
     def validation_step(self, batch: Dict[str, Any], batch_idx: int) -> torch.Tensor:
         """Performs one step of validation on a batch of views."""
         loss = self.loss(batch["views"])
         for k, v in loss.items():
             # Use f-string instead of concatenation
-            self.log(f"val/{k}", v, on_epoch=True, batch_size=batch["views"][0].shape[0])
+            self.log(
+                f"val/{k}", v, on_epoch=True, batch_size=batch["views"][0].shape[0]
+            )
         return loss["objective"]
 
     def test_step(self, batch: Dict[str, Any], batch_idx: int) -> torch.Tensor:
         """Performs one step of testing on a batch of views."""
         loss = self.loss(batch["views"])
         for k, v in loss.items():
             # Use f-string instead of concatenation
-            self.log(f"test/{k}", v, on_epoch=True, batch_size=batch["views"][0].shape[0])
+            self.log(
+                f"test/{k}", v, on_epoch=True, batch_size=batch["views"][0].shape[0]
+            )
         return loss["objective"]
 
     @torch.no_grad()