🎨 Format Python code with psf/black

janelia-cellmap · Aug 5, 2024 · 961ced8 · 961ced8
1 parent a7200c2
commit 961ced8
Showing 1 changed file with 3 additions and 1 deletion.
diff --git a/dacapo/store/local_weights_store.py b/dacapo/store/local_weights_store.py
@@ -142,7 +142,9 @@ def retrieve_weights(self, run: str, iteration: int) -> Weights:
 
         weights_name = self.__get_weights_dir(run) / "iterations" / str(iteration)
 
-        weights: Weights = torch.load(weights_name, map_location="cpu", weights_only=True)
+        weights: Weights = torch.load(
+            weights_name, map_location="cpu", weights_only=True
+        )
         if not isinstance(weights, Weights):
             # backwards compatibility
             weights = Weights(weights["model"], weights["optimizer"])