linting correction

mklasby · Jul 17, 2024 · a1b5b55 · a1b5b55
1 parent 70d7039
commit a1b5b55
Show file tree

Hide file tree

Showing 3 changed files with 5 additions and 3 deletions.
diff --git a/sparsimony/api.py b/sparsimony/api.py
@@ -82,6 +82,7 @@ def set(
         sparsity=sparsity,
     )
 
+
 def static(
         optimizer: torch.optim.Optimizer,
         sparsity: float,

diff --git a/sparsimony/dst/static.py b/sparsimony/dst/static.py
@@ -8,7 +8,7 @@
 from sparsimony.utils import get_mask
 
 
-# TODO - double check if the current default init_method is good to go for static sparsity
+# TODO - double check init_method
 class StaticMagnitudeSparsifier(DSTMixin, BaseSparsifier):
     def __init__(
         self,
@@ -17,7 +17,6 @@ def __init__(
         sparsity: float,
         init_method: str = "sparse_torch",
     ):
-
         optimizer = optimizer
         self.distribution = distribution
         self.sparsity = sparsity

diff --git a/sparsimony/pruners/unstructured.py b/sparsimony/pruners/unstructured.py
@@ -64,7 +64,9 @@ def calculate_mask(
         _, indices = torch.topk(scores.reshape(-1), k=n_drop, largest=False)
         mask = (
             mask.reshape(-1)
-            .scatter(dim=0, index=indices, src=torch.zeros_like(mask.reshape(-1)))
+            .scatter(
+                dim=0, index=indices, src=torch.zeros_like(mask.reshape(-1))
+            )
             .reshape(mask.shape)
         )
         return mask