style: 🎨 Black format.

janelia-cellmap · Dec 5, 2024 · 103ffda · 103ffda
1 parent 6301360
commit 103ffda
Show file tree

Hide file tree

Showing 2 changed files with 5 additions and 3 deletions.
diff --git a/src/cellmap_data/dataloader.py b/src/cellmap_data/dataloader.py
@@ -105,7 +105,7 @@ def refresh(self):
                 "num_workers": self.num_workers,
                 "collate_fn": self.collate_fn,
             }
-        )        
+        )
         if self.sampler is not None:
             if isinstance(self.sampler, Callable):
                 kwargs["sampler"] = self.sampler()

diff --git a/src/cellmap_data/transforms/targets/distance.py b/src/cellmap_data/transforms/targets/distance.py
@@ -103,8 +103,10 @@ def _transform(self, x: torch.Tensor) -> torch.Tensor:
             # return transform_cuda(x) - transform_cuda(x.logical_not())
         else:
             # TODO: Fix this to be correct
-
-            return transform(x).clip(self.clip[0], self.clip[1]) - transform(x.logical_not()).clip(self.clip[0], self.clip[1])
+
+            return transform(x).clip(self.clip[0], self.clip[1]) - transform(
+                x.logical_not()
+            ).clip(self.clip[0], self.clip[1])
 
     def forward(self, x: torch.Tensor) -> torch.Tensor:
         """Forward pass."""