revert changes done by linting.

Author: ParagEkbote

docs/tutorials/index.rst

@@ -73,7 +73,7 @@ These tutorials will guide you through the process of using |pruna| to optimize
    .. grid-item-card:: Smashing at Finer Granularity with Target Modules
       :text-align: center
       :link: ./target_modules_quanto.ipynb
-
+      
       Learn how to use the ``target_modules`` parameter to target specific modules in your model.
 
 .. toctree::
@@ -82,4 +82,4 @@ These tutorials will guide you through the process of using |pruna| to optimize
    :caption: Pruna
    :glob:
 
-   ./*
+   ./*

docs/utils/gen_docs.py

@@ -45,7 +45,9 @@ def generate_algorithm_desc(obj: PrunaAlgorithmBase, name_suffix: str = "") -> s
             f"| **Can be applied on**: {compatible_devices_str}.",
             f"| **Required**: {required_inputs_str}.",
             f"| **Compatible with**: {compatible_algorithms_str}.",
-            f"| **Required install**: {required_install_str}." if required_install_str else "",
+            f"| **Required install**: {required_install_str}."
+            if required_install_str
+            else "",
         ]
     )
 
@@ -81,12 +83,20 @@ def format_grid_table(rows: list[list[str]]) -> str:
     total_widths = [w + 2 for w in col_widths]
 
     horizontal_border = "+" + "+".join("-" * width for width in total_widths) + "+"
-    header_line = "|" + "|".join(" " + rows[0][i].ljust(col_widths[i]) + " " for i in range(num_cols)) + "|"
+    header_line = (
+        "|"
+        + "|".join(" " + rows[0][i].ljust(col_widths[i]) + " " for i in range(num_cols))
+        + "|"
+    )
     header_separator = "+" + "+".join("=" * width for width in total_widths) + "+"
 
     data_lines = []
     for row in rows[1:]:
-        row_line = "|" + "|".join(" " + row[i].ljust(col_widths[i]) + " " for i in range(num_cols)) + "|"
+        row_line = (
+            "|"
+            + "|".join(" " + row[i].ljust(col_widths[i]) + " " for i in range(num_cols))
+            + "|"
+        )
         data_lines.append(row_line)
         data_lines.append(horizontal_border)
 
@@ -212,4 +222,4 @@ def get_table_rows(obj: PrunaAlgorithmBase) -> tuple[list[list[str]], int]:
         for algorithm_group in PRUNA_ALGORITHMS.values():
             for algorithm in algorithm_group.values():
                 f.write(generate_algorithm_desc(algorithm))
-                f.write("\n\n")
+                f.write("\n\n")

src/pruna/evaluation/metrics/metric_elapsed_time.py

@@ -198,11 +198,9 @@ def compute(self, model: PrunaModel, dataloader: DataLoader) -> Dict[str, Any] |
         # Measurement
         list_elapsed_times = []
         with tqdm(total=self.n_iterations, desc="Measuring inference time", unit="iter") as pbar:
-
             def measure_with_progress(m, x):
                 list_elapsed_times.append(self._time_inference(m, x))
                 pbar.update(1)
-
             self._measure(model, dataloader, self.n_iterations, measure_with_progress)
 
         total_elapsed_time = sum(list_elapsed_times)
@@ -349,4 +347,4 @@ def compute(self, model: PrunaModel, dataloader: DataLoader) -> MetricResult:
         # Use EvaluationAgent to share computation across time metrics.
         raw_results = super().compute(model, dataloader)
         result = cast(Dict[str, Any], raw_results)[self.metric_name]
-        return MetricResult(self.metric_name, self.__dict__.copy(), result)
+        return MetricResult(self.metric_name, self.__dict__.copy(), result)

src/pruna/evaluation/metrics/metric_torch.py

@@ -124,7 +124,9 @@ def arniqa_update(metric: ARNIQA, preds: Any) -> None:
 
 
 def ssim_update(
-    metric: StructuralSimilarityIndexMeasure | MultiScaleStructuralSimilarityIndexMeasure, preds: Any, target: Any
+        metric: StructuralSimilarityIndexMeasure | MultiScaleStructuralSimilarityIndexMeasure,
+        preds: Any,
+        target: Any
 ) -> None:
     """
     Update handler for SSIM or MS-SSIM metric.
@@ -400,4 +402,4 @@ def get_call_type(call_type: str, metric_name: str) -> str:
             return get_single_pairing(TorchMetrics[metric_name].call_type)
     else:
         pruna_logger.error(f"Invalid call type: {call_type}. Must be one of {CALL_TYPES}.")
-        raise ValueError(f"Invalid call type: {call_type}. Must be one of {CALL_TYPES}.")
+        raise ValueError(f"Invalid call type: {call_type}. Must be one of {CALL_TYPES}.")