Skip to content

Commit 458fe52

Browse files
Merge pull request #200 from PhilMiller/198-cli
Fix cli execution
2 parents 8cdc429 + 25044bb commit 458fe52

File tree

1 file changed

+5
-4
lines changed

1 file changed

+5
-4
lines changed

src/cli.py

Lines changed: 5 additions & 4 deletions
Original file line numberDiff line numberDiff line change
@@ -9,8 +9,8 @@
99
from pandas import DataFrame
1010

1111
from penn_chime.parameters import Parameters
12-
from penn_chime.utils import build_admissions_df, build_census_df, RateLos
13-
12+
from penn_chime.utils import RateLos
13+
from penn_chime.models import build_admissions_df, build_census_df
1414

1515
class FromFile(Action):
1616
"""From File."""
@@ -123,8 +123,9 @@ def main():
123123
"Recovered": p.recovered_v,
124124
}
125125
)
126-
admits_df = build_admissions_df(p.n_days, *p.dispositions)
127-
census_df = build_census_df(admits_df, *p.lengths_of_stay)
126+
127+
admits_df = build_admissions_df(p)
128+
census_df = build_census_df(admits_df, p)
128129

129130
prefix = a.prefix
130131
for df, name in (

0 commit comments

Comments
 (0)