fix mypy typing errors

pph-collective · Feb 15, 2023 · 7fa17a4 · 7fa17a4
1 parent a27b361
commit 7fa17a4
Show file tree

Hide file tree

Showing 2 changed files with 4 additions and 4 deletions.
diff --git a/titan/features/high_risk.py b/titan/features/high_risk.py
@@ -128,7 +128,7 @@ def set_stats(self, stats: Dict[str, int], time: int):
     # ============== HELPER METHODS ================
 
     def become_high_risk(
-        self, pop: "population.Population", time: int, duration: int = None
+        self, pop: "population.Population", time: int, duration: Optional[int] = None
     ):
         """
         Mark an agent as high risk and assign a duration to their high risk period

diff --git a/titan/model.py b/titan/model.py
@@ -192,9 +192,9 @@ def step(self, outdir: str):
             "  STARTING HIV count:{}  Total Incarcerated:{}  HR+:{}  "
             "PrEP:{}".format(
                 len(exposures.HIV.agents),
-                sum([1 for a in self.pop.all_agents if a.incar.active]),  # type: ignore[attr-defined]
-                sum([1 for a in self.pop.all_agents if a.high_risk.active]),  # type: ignore[attr-defined]
-                sum([1 for a in self.pop.all_agents if a.prep.active]),  # type: ignore[attr-defined]
+                sum([1 for a in self.pop.all_agents if a.incar.active]),  # type: ignore[misc, attr-defined]
+                sum([1 for a in self.pop.all_agents if a.high_risk.active]),  # type: ignore[misc, attr-defined]
+                sum([1 for a in self.pop.all_agents if a.prep.active]),  # type: ignore[misc, attr-defined]
             )
         )