Fix mypy errors

se-sic · Jul 25, 2023 · b8a257e · b8a257e
1 parent 1943d32
commit b8a257e
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/varats/varats/plots/llvm_coverage_plot.py b/varats/varats/plots/llvm_coverage_plot.py
@@ -142,7 +142,7 @@ def _compute_confusion_matrix(
     feature_report: CoverageReport,
     feature_name_map: tp.Dict[str, str],
     threshold: float = 1.0
-) -> ConfusionMatrix:
+) -> ConfusionMatrix[ConfusionEntry]:
     coverage_feature_regions: tp.List[tp.Any] = []
     coverage_normal_regions: tp.List[tp.Any] = []
     vara_feature_regions: tp.List[tp.Any] = []
@@ -194,7 +194,7 @@ def confusion_matrices(
         self,
         feature_name_map: tp.Dict[str, str],
         threshold: float = 1.0
-    ) -> tp.Dict[str, ConfusionMatrix]:
+    ) -> tp.Dict[str, ConfusionMatrix[ConfusionEntry]]:
         """Returns the confusion matrices."""
 
         report = self.feature_report()