ruff format

metrics/benchmarks/core/models.py

@@ -1,16 +1,11 @@
-__all__ = [
-    "Benchmark",
-    "Benchmarks",
-    "BenchmarkResult",
-    "BenchmarkGroupResult"
-]
+__all__ = ["Benchmark", "Benchmarks", "BenchmarkResult", "BenchmarkGroupResult"]
 
+import gc
 import io
+import statistics
+import time
 from contextlib import redirect_stdout
 from dataclasses import dataclass
-import time
-import gc
-import statistics
 from typing import Callable, override
 
 from .exceptions import BenchmarkNotFound, BenchmarksNotFound, BenchmarksWithSameNameAlreadyExists
@@ -40,14 +35,7 @@ class BenchmarkGroupResult:
 
 
 class Benchmark:
-    def __init__(
-            self,
-            func: FuncForBenchmark,
-            *,
-            type_: str,
-            name: str,
-            description: str
-    ) -> None:
+    def __init__(self, func: FuncForBenchmark, *, type_: str, name: str, description: str) -> None:
         self.func = func
         self.type_ = type_
         self.name = name
@@ -78,11 +66,11 @@ class Benchmark:
 
     @override
     def __repr__(self) -> str:
-        return f'Benchmark<{self.type_=}, {self.name=}, {self.description=}>'
+        return f"Benchmark<{self.type_=}, {self.name=}, {self.description=}>"
 
     @override
     def __str__(self) -> str:
-        return f'benchmark {self.name} with type {self.type_}'
+        return f"benchmark {self.name} with type {self.type_}"
 
 
 class Benchmarks:
@@ -92,16 +80,14 @@ class Benchmarks:
         self._benchmarks_paired_by_name: dict[str, Benchmark] = {}
 
     def register(
-            self,
-            type_: str,
-            description: str = ""
+        self, type_: str, description: str = ""
     ) -> Callable[[FuncForBenchmark], FuncForBenchmark]:
         def decorator(func: FuncForBenchmark) -> FuncForBenchmark:
             benchmark = Benchmark(
                 func,
                 type_=type_,
                 name=func.__name__,
-                description=description or f'description for {func.__name__} with type {type_}',
+                description=description or f"description for {func.__name__} with type {type_}",
             )
             if self._benchmarks_paired_by_name.get(func.__name__):
                 raise BenchmarksWithSameNameAlreadyExists(func.__name__)
@@ -110,9 +96,12 @@ class Benchmarks:
             self._benchmarks.append(benchmark)
             self._benchmarks_grouped_by_type.setdefault(type_, []).append(benchmark)
             return func
+
         return decorator
 
-    def run_benchmark_by_name(self, name: str, iterations: int = 100, is_gc_disables: bool = False) -> BenchmarkResult:
+    def run_benchmark_by_name(
+        self, name: str, iterations: int = 100, is_gc_disables: bool = False
+    ) -> BenchmarkResult:
         benchmark = self.get_benchmark_by_name(name)
         if not benchmark:
             raise BenchmarkNotFound(name)
@@ -130,28 +119,34 @@ class Benchmarks:
             is_gc_disabled=is_gc_disables,
             avg_time=avg,
             median_time=median,
-            std_dev=std_dev
+            std_dev=std_dev,
         )
 
-    def run_benchmarks_by_type(self, type_: str, iterations: int = 100, is_gc_disabled: bool = False) -> BenchmarkGroupResult:
+    def run_benchmarks_by_type(
+        self, type_: str, iterations: int = 100, is_gc_disabled: bool = False
+    ) -> BenchmarkGroupResult:
         benchmarks = self.get_benchmarks_by_type(type_)
         if not benchmarks:
             raise BenchmarksNotFound(type_)
         benchmark_results: list[BenchmarkResult] = []
 
         for benchmark in benchmarks:
-            benchmark_results.append(self.run_benchmark_by_name(benchmark.name, iterations, is_gc_disabled))
+            benchmark_results.append(
+                self.run_benchmark_by_name(benchmark.name, iterations, is_gc_disabled)
+            )
 
         return BenchmarkGroupResult(
             type_=type_,
             iterations=iterations,
             is_gc_disabled=is_gc_disabled,
-            benchmark_results=benchmark_results
+            benchmark_results=benchmark_results,
         )
 
-    def run_benchmarks_grouped_by_type(self, iterations: int = 100, is_gc_disabled: bool = False) -> list[BenchmarkGroupResult]:
+    def run_benchmarks_grouped_by_type(
+        self, iterations: int = 100, is_gc_disabled: bool = False
+    ) -> list[BenchmarkGroupResult]:
         results: list[BenchmarkGroupResult] = []
-        for type_, benchmarks in self._benchmarks_grouped_by_type.items():
+        for type_, _ in self._benchmarks_grouped_by_type.items():
             results.append(self.run_benchmarks_by_type(type_, iterations, is_gc_disabled))
         return results