black

rwth-i6 · Dec 4, 2024 · 40c369b · 40c369b
1 parent b50b66e
commit 40c369b
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/i6_models/parts/best_rq/mask.py b/i6_models/parts/best_rq/mask.py
@@ -67,7 +67,7 @@ def forward(
             mask_idc = np.random.choice(seq_len - min_len, num_mask, replace=False)
 
             for j in mask_idc:
-                mask[i, j : j+self.mask_length] = True
+                mask[i, j : j + self.mask_length] = True
 
         tensor[mask] = self.mask_emb.to(tensor.device)