Update to ruff format II

Author: palonso

src/nets/rope.py

@@ -53,9 +53,9 @@ def apply_rotary_emb(freqs, t, start_index=0, scale=1.0, seq_dim=-2):
     rot_dim = freqs.shape[-1]
     end_index = start_index + rot_dim
 
-    assert (
-        rot_dim <= t.shape[-1]
-    ), f"feature dimension {t.shape[-1]} is not of sufficient size to rotate in all the positions {rot_dim}"
+    assert rot_dim <= t.shape[-1], (
+        f"feature dimension {t.shape[-1]} is not of sufficient size to rotate in all the positions {rot_dim}"
+    )
 
     t_left, t, t_right = (
         t[..., :start_index],
@@ -173,9 +173,9 @@ def get_seq_pos(self, seq_len, device, dtype, offset=0):
     def rotate_queries_or_keys(self, t, seq_dim=None, offset=0, scale=None):
         seq_dim = default(seq_dim, self.default_seq_dim)
 
-        assert (
-            not self.use_xpos or exists(scale)
-        ), "you must use `.rotate_queries_and_keys` method instead and pass in both queries and keys, for length extrapolatable rotary embeddings"
+        assert not self.use_xpos or exists(scale), (
+            "you must use `.rotate_queries_and_keys` method instead and pass in both queries and keys, for length extrapolatable rotary embeddings"
+        )
 
         device, dtype, seq_len = t.device, t.dtype, t.shape[seq_dim]
 

src/prediction/dataset.py

@@ -41,9 +41,9 @@ def __init__(
         self.n_frames = num_frames
         self.n_seconds = 30
 
-        assert (
-            self.overlap_ratio >= 0 and self.overlap_ratio < 1
-        ), "Overlap ratio must be between 0 and 1."
+        assert self.overlap_ratio >= 0 and self.overlap_ratio < 1, (
+            "Overlap ratio must be between 0 and 1."
+        )
 
         self.compute_segments_per_file()
 

src/probe/modules/sequence_classifiers.py

@@ -195,7 +195,7 @@ def plot_confusion_matrix(self, conf_matrix, multiclass=False):
             )
             axes = axes.flatten()
             labels = (
-                [f"{i+1}" for i in range(50)] if self.labels is None else self.labels
+                [f"{i + 1}" for i in range(50)] if self.labels is None else self.labels
             )
             for ax, cm, label in zip(axes, conf_matrix, labels):
                 # Plot the confusion matrix in each subplot