add missing type hint

mlcommons · Dec 8, 2023 · bd8c471 · bd8c471
1 parent 0acfa07
commit bd8c471
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/src/coffee/run.py b/src/coffee/run.py
@@ -68,7 +68,7 @@ def add(self, test, sut, test_sut_scores):
         self.data[(test.__class__.__name__, sut)].append(test_sut_scores)
 
     def for_sut(self, desired_sut) -> dict:
-        result = defaultdict(dict)
+        result: defaultdict = defaultdict(dict)
         for test, sut in self.data:
             if sut == desired_sut:
                 for entry in self.data[(test, sut)]: