@@ -15,23 +15,24 @@ def run_module(params):
15
15
The parameters dictionary must include the signals and signals.
16
16
We are only considering lag-1 data.
17
17
"""
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 } " )])
28
29
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