CLN: add typing to autotimeit.py

pydata · Jan 18, 2020 · 37110cf · 37110cf
1 parent 33d4bac
commit 37110cf
Showing 1 changed file with 6 additions and 3 deletions.
diff --git a/bottleneck/benchmark/autotimeit.py b/bottleneck/benchmark/autotimeit.py
@@ -1,16 +1,19 @@
 import timeit
+from typing import Tuple
 
 
-def autotimeit(stmt, setup="pass", repeat=3, mintime=0.2):
+def autotimeit(
+    stmt: str, setup: str = "pass", repeat: int = 3, mintime: float = 0.2
+) -> float:
     timer = timeit.Timer(stmt, setup)
     number, time1 = autoscaler(timer, mintime)
     time2 = timer.repeat(repeat=repeat - 1, number=number)
     return min(time2 + [time1]) / number
 
 
-def autoscaler(timer, mintime):
+def autoscaler(timer: timeit.Timer, mintime: float) -> Tuple[int, float]:
     number = 1
-    for i in range(12):
+    for _ in range(12):
         time = timer.timeit(number)
         if time > mintime:
             return number, time