Merge remote-tracking branch 'origin/main' into fix_whiten

Merge flake8 fixes.
rctn · Dec 19, 2023 · 64fe099 · 64fe099
2 parents 8d1aac0 + 16eb930
commit 64fe099
Show file tree

Hide file tree

Showing 2 changed files with 4 additions and 4 deletions.
diff --git a/sparsecoding/models.py b/sparsecoding/models.py
@@ -26,9 +26,9 @@ def __init__(self, inference_method, n_basis, n_features,
         device : torch.device, default=torch.device("cpu")
             Which device to utilize
         check_for_dictionary_nan : bool, default=False
-            Flag to check for nans in the dictionary after gradient 
-            updates and normalizations. Raises ValueError if nan 
-            found 
+            Flag to check for nans in the dictionary after gradient
+            updates and normalizations. Raises ValueError if nan
+            found
         """
         super(SparseCoding, self).__init__()
         self.inference_method = inference_method

diff --git a/tutorials/vanilla/src/utils.py b/tutorials/vanilla/src/utils.py
@@ -145,7 +145,7 @@ def create_patches(imgs, epochs, batch_size, N, rng):
     patches : Tensor of size (epochs, batch_size, pixels_per_patch).
     """
     # TODO: use rng here when sample_random_patches supports it.
-    patches = sample_random_patches(int(np.sqrt(N)), batch_size*epochs, 
+    patches = sample_random_patches(int(np.sqrt(N)), batch_size*epochs,
                                     torch.unsqueeze(imgs, 1))
     patches = patches.reshape(epochs, batch_size, N)
     return patches