Skip to content

Commit f5d15e8

Browse files
Ananya JoshiAnanya Joshi
Ananya Joshi
authored and
Ananya Joshi
committed
changed out-of-range handling and params for run when flash not part of the json
1 parent f3b022a commit f5d15e8

File tree

2 files changed

+21
-21
lines changed

2 files changed

+21
-21
lines changed

_delphi_utils_python/delphi_utils/flash_eval/eval_day.py

Lines changed: 1 addition & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -274,8 +274,7 @@ def flash_eval(lag, day, input_df, signal, params, logger=None):
274274

275275
# discuss where to do out-of-range handling
276276
out_range = input_df.columns[input_df.lt(int(params['flash']['support'][0])).iloc[0, :].values
277-
& input_df.gt(int(params['flash']['support'][1])).iloc[0, :].values ]
278-
277+
| input_df.gt(int(params['flash']['support'][1])).iloc[0, :].values ]
279278

280279
#only rank streams without out of range data
281280
input_df = input_df[filter(lambda x: x not in out_range, input_df.columns)]

_delphi_utils_python/delphi_utils/flash_eval/run.py

Lines changed: 20 additions & 19 deletions
Original file line numberDiff line numberDiff line change
@@ -15,23 +15,24 @@ def run_module(params):
1515
The parameters dictionary must include the signals and signals.
1616
We are only considering lag-1 data.
1717
"""
18-
signals = params["flash"]["signals"]
19-
for signal in signals:
20-
export_files = read_filenames(params["common"]["export_dir"])
21-
days = {}
22-
for (x, _) in export_files:
23-
#Concat the data from recent files at nation, state, and county resolution per day.
24-
if signal in x and pd.Series([y in x for y in ['state', 'county', 'nation']]).any():
25-
day = pd.to_datetime(x.split('_')[0], format="%Y%m%d", errors='raise')
26-
days[day] = pd.concat([days.get(day, pd.DataFrame()),
27-
load_csv(f"{params['common']['export_dir']}/{x}")])
18+
if params.get("flash", None):
19+
signals = params["flash"].get("signals", [])
20+
for signal in signals:
21+
export_files = read_filenames(params["common"]["export_dir"])
22+
days = {}
23+
for (x, _) in export_files:
24+
#Concat the data from recent files at nation, state, and county resolution per day.
25+
if signal in x and pd.Series([y in x for y in ['state', 'county', 'nation']]).any():
26+
day = pd.to_datetime(x.split('_')[0], format="%Y%m%d", errors='raise')
27+
days[day] = pd.concat([days.get(day, pd.DataFrame()),
28+
load_csv(f"{params['common']['export_dir']}/{x}")])
2829

29-
for day, input_df in dict(sorted(days.items())).items():
30-
input_df = input_df[['geo_id', 'val']].set_index('geo_id').T
31-
input_df.index = [day]
32-
today = date.today()
33-
lag= (pd.to_datetime(today)-pd.to_datetime(day)).days
34-
# inital flash implementation assume lag == 1 always
35-
#if str(lag) in params["flash"]["lags"]:
36-
lag=1
37-
flash_eval(int(lag), day, input_df, signal, params)
30+
for day, input_df in dict(sorted(days.items())).items():
31+
input_df = input_df[['geo_id', 'val']].set_index('geo_id').T
32+
input_df.index = [day]
33+
today = date.today()
34+
lag= (pd.to_datetime(today)-pd.to_datetime(day)).days
35+
# inital flash implementation assume lag == 1 always
36+
#if str(lag) in params["flash"]["lags"]:
37+
lag=1
38+
flash_eval(int(lag), day, input_df, signal, params)

0 commit comments

Comments
 (0)