Add flu hospitalizations

Author: krivard

hhs_hosp/delphi_hhs/constants.py

@@ -5,12 +5,16 @@
 SUM_CONF_SUSP = "sum_confirmed_suspected_admissions_covid_1d"
 CONFIRMED_PROP = "confirmed_admissions_covid_1d_prop"
 SUM_CONF_SUSP_PROP = "sum_confirmed_suspected_admissions_covid_1d_prop"
+CONFIRMED_FLU = "confirmed_admissions_influenza_1d"
+CONFIRMED_FLU_PROP = CONFIRMED_FLU+"_prop"
 
 SIGNALS = [
     CONFIRMED,
     SUM_CONF_SUSP,
     CONFIRMED_PROP,
-    SUM_CONF_SUSP_PROP
+    SUM_CONF_SUSP_PROP,
+    CONFIRMED_FLU,
+    CONFIRMED_FLU_PROP
 ]
 
 GEOS = [

hhs_hosp/delphi_hhs/run.py

@@ -15,7 +15,7 @@
 import numpy as np
 import pandas as pd
 
-from .constants import SIGNALS, GEOS, SMOOTHERS, CONFIRMED, SUM_CONF_SUSP
+from .constants import SIGNALS, GEOS, SMOOTHERS, CONFIRMED, SUM_CONF_SUSP, CONFIRMED_FLU
 
 def _date_to_int(d):
     """Return a date object as a yyyymmdd int."""
@@ -190,6 +190,13 @@ def make_signal(all_columns, sig):
             all_columns.previous_day_admission_pediatric_covid_confirmed + \
             all_columns.previous_day_admission_pediatric_covid_suspected,
         })
+    elif sig.startswith(CONFIRMED_FLU):
+        df = pd.DataFrame({
+            "state": all_columns.state.apply(str.lower),
+            "timestamp":int_date_to_previous_day_datetime(all_columns.date),
+            "val": \
+            all_columns.previous_day_admission_influenza_confirmed
+        })
     else:
         raise Exception(
             "Bad programmer: signal '{sig}' in SIGNALS but not handled in make_signal"