line was too long

ruflab · Jul 30, 2024 · a5b26a8 · a5b26a8
1 parent 61d0c7e
commit a5b26a8
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/shimmer/modules/selection.py b/shimmer/modules/selection.py
@@ -216,7 +216,7 @@ def __init__(
             domain_dim (`int`) : dimension of the input dims (assumed to be the same
                 for now)
             domain_names  (`Iterable[str]`) : list of input domains
-            n_steps (`int`) : number of steps to update the query vector, where 0 steps is static attention.
+            n_steps (`int`) : number of steps to update the query vector
         """
         super().__init__()
         self.head_size = head_size