fix: format warning

aim-qmul · May 16, 2023 · 0202cf3 · 0202cf3
1 parent f6a71b9
commit 0202cf3
Showing 1 changed file with 3 additions and 1 deletion.
diff --git a/data/augment.py b/data/augment.py
@@ -430,7 +430,9 @@ def _transform(self, x: torch.Tensor):
         num_segments = randint(1, max_num_segments)
         segment_lengths = (
             x.shape[-1]
-            * np.random.dirichlet([rand(0, 10) for _ in range(num_segments)], 1)  # TODO(cm): this can crash training
+            * np.random.dirichlet(
+                [rand(0, 10) for _ in range(num_segments)], 1
+            )  # TODO(cm): this can crash training
         ).astype("int")[0]
 
         segment_lengths = np.maximum(segment_lengths, 1)