style: ruff formatting

Author: rensortino

src/pruna/algorithms/token_merging.py

@@ -248,9 +248,7 @@ def forward(
                 attn_weights = attn_weights * head_mask
 
             attn_weights = attn_weights.softmax(dim=-1)
-            attn_probs = torch.nn.functional.dropout(
-                attn_weights, p=self.dropout_prob if self.training else 0.0
-            )
+            attn_probs = torch.nn.functional.dropout(attn_weights, p=self.dropout_prob if self.training else 0.0)
 
             context_layer = (attn_probs @ value_layer).transpose(1, 2)
             context_layer = context_layer.reshape(batch_size, -1, self.all_head_size)

src/pruna/engine/model_checks.py

@@ -140,7 +140,7 @@ def is_vit(model: Any) -> bool:
     -------
     bool
         True if the model is a ViT model, False otherwise.
-     """
+    """
     return model.__class__.__name__ == "ViTForImageClassification"
 
 
@@ -160,6 +160,7 @@ def is_transformers_pipeline_with_vit(model: Any) -> bool:
     """
     return isinstance(model, ImageClassificationPipeline) and is_vit(getattr(model, "model", None))
 
+
 def is_transformers_pipeline_with_causal_lm(model: Any) -> bool:
     """
     Check if the model is a transformers pipeline (for tasks like text generation, classification, etc.).