Merge pull request #59 from rctn/fix_flake9

fix flake8 errors
rctn · Dec 19, 2023 · 16eb930 · 16eb930
2 parents ef5eb03 + 835b9e8
commit 16eb930
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