Format

RasmussenLab · Jan 8, 2025 · 1b4a7af · 1b4a7af
1 parent 0b6971c
commit 1b4a7af
Show file tree

Hide file tree

Showing 3 changed files with 9 additions and 3 deletions.
diff --git a/vamb/encode.py b/vamb/encode.py
@@ -514,7 +514,9 @@ def load(
         """
 
         # Forcably load to CPU even if model was saves as GPU model
-        dictionary = _torch.load(path, map_location=lambda storage, loc: storage, weights_only=True)
+        dictionary = _torch.load(
+            path, map_location=lambda storage, loc: storage, weights_only=True
+        )
 
         nsamples = dictionary["nsamples"]
         alpha = dictionary["alpha"]

diff --git a/vamb/semisupervised_encode.py b/vamb/semisupervised_encode.py
@@ -1125,7 +1125,9 @@ def load(cls, path, cuda=False, evaluate=True):
         """
 
         # Forcably load to CPU even if model was saves as GPU model
-        dictionary = _torch.load(path, map_location=lambda storage, loc: storage, weights_only=False)
+        dictionary = _torch.load(
+            path, map_location=lambda storage, loc: storage, weights_only=False
+        )
 
         nsamples = dictionary["nsamples"]
         nlabels = dictionary["nlabels"]

diff --git a/vamb/taxvamb_encode.py b/vamb/taxvamb_encode.py
@@ -630,7 +630,9 @@ def load(cls, path, nodes, table_parent, cuda=False, evaluate=True):
         """
 
         # Forcably load to CPU even if model was saves as GPU model
-        dictionary = _torch.load(path, map_location=lambda storage, loc: storage, weights_only=False)
+        dictionary = _torch.load(
+            path, map_location=lambda storage, loc: storage, weights_only=False
+        )
 
         nsamples = dictionary["nsamples"]
         nlabels = dictionary["nlabels"]