Corrected types.

plasma-umass · Nov 4, 2024 · 024e8ff · 024e8ff
1 parent 0c71622
commit 024e8ff
Showing 1 changed file with 3 additions and 3 deletions.
diff --git a/scalene/scalene_leak_analysis.py b/scalene/scalene_leak_analysis.py
@@ -1,4 +1,4 @@
-from typing import Any, List, OrderedDict
+from typing import Any, List, OrderedDict, Tuple
 
 from scalene.scalene_statistics import Filename, LineNumber, ScaleneStatistics
 
@@ -17,10 +17,10 @@ def compute_leaks(
         stats: ScaleneStatistics,
         avg_mallocs: OrderedDict[LineNumber, float],
         fname: Filename,
-    ) -> List[Any]:
+    ) -> List[Tuple[LineNumber, float, float]]:
         if growth_rate / 100 < ScaleneLeakAnalysis.growth_rate_threshold:
             return []
-        leaks = []
+        leaks : List[Tuple[LineNumber, float, float]] = []
         keys = list(stats.leak_score[fname].keys())
         for index, item in enumerate(stats.leak_score[fname].values()):
             # See https://en.wikipedia.org/wiki/Rule_of_succession