demographic_columns = [
    "dob",
    "age",
    "sex",
    "hispanic",
    "race",
    "residence",
    "hcw",
    "smoker",
    "bmi",
]

date_columns = [
    "presentation_date1",
    "discharge_date1",
    "admission_date2",
    "discharge_date2",
    "icu_admission_date_1",
    "icu_admission_date_2",
    'icu_discharge_date_1',
    "icu_discharge_date_2",
    "covid_collection_date_1",
    "covid_result_date_1",
    "covid_collection_date_2",
    "covid_result_date_2",
    "covid_collection_date_3",
    "covid_result_date_3",
    "intubation_date_1",
    "extubation_date_1",
    "intubation_date_2",
    "extubation_date_2",
    "death_date",
    "wbc_admission_date",
    "creatinine_admission_date",
    "bun_admission_date",
    "ast_admission_date",
    "ldh_admission_date",
    "crp_admission_date",
    "ddimer_admission_date",
    "inr_admission_date",
    "fibrinogen_admission_date",
    "ferritin_admission_date",
    "ptt_admission_date",
    "cd4_date",
]

status_columns = [
  "admission_level",
  "discharge_destination_1",
  "discharge_destination_2",
  "days_of_symptoms_before_1st_admission",
  "still_inpatient", 
]

symptoms_columns = [
    "fever",
    "cough",
    "shortness_of_breath",
    "fatigue",
    "sore_throat",
    "chills",
    "headache",
    "diarrhea",
    "other_symptoms",
    "fever_duration", 
]

vitals_columns = [
    "initial_temp",
    "max_temp_admission",
    "systolic_bp",
    "diastolic_bp",
    "initial_rr",
    "o2_saturation",
    "lung_exam",
    "altered_mental_status",
]

categorical_events_columns = [
    "antipyretics",
    "received_oral_meds_in_1st_24hr",
    "on_o2_at_presentation",
    "intubation",
    "hi_flow",
    "non_rebreather_mask",
    "cpap_bpap",
    "nasal_cannula",
    "aki",
    "concurrent_pathogen_infection",
    "sepsis",
    "cardiogenic_shock",
    "myocarditis",
    "cardiac_arrest",
    "ards",
    "ecmo",
    "other_complications",
    "readmission",
    "composite",
    "death",
    'covid_result_1',
    'covid_result_2',
    'covid_result_3',    
]

radiology_columns = [
    "admission_xray",
    "cxr_bilateral_opacities",
    "cxr_interstitial",
    "cxr_unilateral_opacities",
    "initial_lung_ultrasound",
    "initial_heart_ultrasound",    
]

comorbidities_columns = [
    "copd",
    "asthma",
    "hypertension",
    "diabetes",
    "pulmonary_dx",
    "heart_disease",
    "malignancy",
    "ckd",
    "hd",
    "immunocompromised",
    "hiv",
    "other_comorbidity_bool",
    "other_comorbidity",
    "ace",
    "arb",
]

treatments_columns = [
    "chloroquine_days",
    "hydroxychloroquine_days",
    "kaletra_days",
    "remdesivir_compassionate_days",
    "remdesivir_clinical_trial_days",
    "other_trial",
    "pressors_days",
    "pronation_treatment",
    "other_iv_abx",
    "azithromycin_days",
    "glucose_6",
    "glucocorticoids",
    "nsaid",
]

labs_columns = [
    "a1c",
    "cd4_count_most_recent",
    "cd4_percent_most_recent",
    "wbcadmac",
    "wbcpeakac",
    "wbcnadirac",
    "crpeakac",
    "cradmac",
    "bunadmac",
    "neutadmac",
    "neutnadirac",
    "lymphadmac",
    "lymphnadirac",
    "pltpeakac",
    "pltnadirac",
    "pltadmac",
    "astpeakac",
    "astadmac",
    "astadmac",
    "altadmac",
    "ldhpeakac",
    "ldhadmac",
    "procalpeakac",
    "procaladmac",
    "crppeakac",
    "crpadmac",
    "ddimeradmac",
    "ddimerpeakac",
    "inradmac",
    "inrpeakac",
    "fibrinoadmac",
    "fibrinopeakac",
    "ferritadmac",
    "ferritpeakac",
    "pttadmac",
    "pttpeakac",
    "ptadmac",
    "ptpeakac",
    "alkpadmac",
    "alkppeakac",
    "albadmac",
    "il6initac",
    "il6peakac",
    "eosadmac",
    "g6pd-def1-norml0-inter2",
    "bpsystac",
    "bpdiastac",
    "tempinitac",
    "tempmaxac",
    "rrinitac",
    "pulseoxinitac",
]

studies = [
 'AH',
 'AK',
 'BH',
 'DB',
 'EC',
 'EH',
 'ET',
 'IL',
 'JQ',
 'MK',
 'NP',
 'SA',
 'YV',
 'no study',
]