Add ppe csv to cli
@@ -6,7 +6,9 @@ import sys
from .model.parameters import Parameters
from .model.sir import Sir
+
def run(argv):
+ """Eun cli."""
p = Parameters.create(os.environ, argv[1:])
m = Sir(p)
@@ -14,9 +16,11 @@ def run(argv):
(m.sim_sir_w_date_df, "sim_sir_w_date"),
(m.admits_df, "projected_admits"),
(m.census_df, "projected_census"),
+ (m.ppe_df, 'ppe_data')
):
df.to_csv(f"{p.current_date}_{name}.csv")
+
def main():
"""Main."""
run(sys.argv)
