Skip to content

Commit 7dc652e

Browse files
committed
Delete silly intermediate function
1 parent 010d3e8 commit 7dc652e

File tree

1 file changed

+2
-11
lines changed

1 file changed

+2
-11
lines changed

src/penn_chime/models.py

Lines changed: 2 additions & 11 deletions
Original file line numberDiff line numberDiff line change
@@ -265,10 +265,10 @@ def sir(
265265
return s_n * scale, i_n * scale, r_n * scale
266266

267267

268-
def gen_sir(
268+
def sim_sir(
269269
s: float, i: float, r: float, gamma: float, i_day: int, policies: Sequence[Tuple[float, int]]
270270
):
271-
"""Simulate SIR model forward in time yielding tuples.
271+
"""Simulate SIR model forward in time, returning a dictionary of daily arrays
272272
Parameter order has changed to allow multiple (beta, n_days)
273273
to reflect multiple changing social distancing policies.
274274
"""
@@ -310,15 +310,6 @@ def gen_sir(
310310
}
311311

312312

313-
def sim_sir(
314-
s: float, i: float, r: float,
315-
gamma: float, i_day: int, policies: Sequence[Tuple[float, int]]
316-
) -> pd.DataFrame:
317-
"""Simulate the SIR model forward in time."""
318-
data = gen_sir(s, i, r, gamma, i_day, policies)
319-
return data
320-
321-
322313
def build_sim_sir_w_date_df(
323314
raw_df: pd.DataFrame,
324315
current_date: datetime,

0 commit comments

Comments
 (0)