Fix type hints

hajimes · Aug 22, 2024 · 2092d53 · 2092d53
1 parent ea50371
commit 2092d53
Showing 1 changed file with 3 additions and 2 deletions.
diff --git a/benchmark/benchmark.py b/benchmark/benchmark.py
@@ -1,7 +1,7 @@
 import gc
 import hashlib
 import time
-from collections.abc import Callable, List
+from typing import Any, Callable, Dict, List
 
 import matplotlib.pyplot as plt
 import mmh3
@@ -55,7 +55,7 @@ def benchmark_function(self, params):
 
         return result
 
-    def run_timed_benchmarks(self, params):
+    def run_timed_benchmarks(self, params: Dict[Any]):
         time_spent = 0
 
         WOLKLOAD_MULTIPLIER = 10
@@ -190,6 +190,7 @@ def benchmark_throughput_small_inputs(
 
     return data_result
 
+
 def benchmark_large_inputs(
     hashes: List[Callable], large_test_log_min: int, large_test_log_max: int
 ):