Skip to content

Commit 7aca5d2

Browse files
committed
Clear up typing over-concreteness
1 parent e7801a8 commit 7aca5d2

File tree

1 file changed

+5
-5
lines changed

1 file changed

+5
-5
lines changed

src/penn_chime/models.py

Lines changed: 5 additions & 5 deletions
Original file line numberDiff line numberDiff line change
@@ -9,7 +9,7 @@
99
from datetime import date, datetime, timedelta
1010
from logging import INFO, basicConfig, getLogger
1111
from sys import stdout
12-
from typing import Dict, Generator, Tuple, Sequence,Optional
12+
from typing import Dict, Generator, Tuple, Sequence, Optional
1313

1414
import numpy as np
1515
import pandas as pd
@@ -145,7 +145,7 @@ def __init__(self, p: Parameters):
145145
self.daily_growth_rate = get_growth_rate(p.doubling_time)
146146
self.daily_growth_rate_t = get_growth_rate(self.doubling_time_t)
147147

148-
def gen_policy(self, p: Parameters) -> List[Tuple[float, int]]:
148+
def gen_policy(self, p: Parameters) -> Sequence[Tuple[float, int]]:
149149
if p.mitigation_date is not None:
150150
mitigation_day = -(p.current_date - p.mitigation_date).days
151151
else:
@@ -164,7 +164,7 @@ def gen_policy(self, p: Parameters) -> List[Tuple[float, int]]:
164164
(self.beta_t, post_mitigation_days),
165165
]
166166

167-
def run_projection(self, p: Parameters, policy: List[Tuple[float, int]]):
167+
def run_projection(self, p: Parameters, policy: Sequence[Tuple[float, int]]):
168168
self.raw_df = sim_sir_df(
169169
self.susceptible,
170170
self.infected,
@@ -237,7 +237,7 @@ def sir(
237237

238238

239239
def gen_sir(
240-
s: float, i: float, r: float, gamma: float, i_day: int, policies: List[Tuple[float, int]]
240+
s: float, i: float, r: float, gamma: float, i_day: int, policies: Sequence[Tuple[float, int]]
241241
) -> Generator[Tuple[int, float, float, float], None, None]:
242242
"""Simulate SIR model forward in time yielding tuples.
243243
Parameter order has changed to allow multiple (beta, n_days)
@@ -256,7 +256,7 @@ def gen_sir(
256256

257257
def sim_sir_df(
258258
s: float, i: float, r: float,
259-
gamma: float, i_day: int, policies: List[Tuple[float, int]]
259+
gamma: float, i_day: int, policies: Sequence[Tuple[float, int]]
260260
) -> pd.DataFrame:
261261
"""Simulate the SIR model forward in time."""
262262
return pd.DataFrame(

0 commit comments

Comments
 (0)