fixing madspeedfilder

max_dilation speed was initialized as np.empty but should be an array of np.nans
nih-megcore · Mar 5, 2024 · bf5db9f · bf5db9f
1 parent 81b50f9
commit bf5db9f
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/nih2mne/eyetracking_preprocessing.py b/nih2mne/eyetracking_preprocessing.py
@@ -137,7 +137,7 @@ def madspeedfilter(tv,dia,is_valid):
     fwd_dilation                                = np.pad(cur_dia_speed,(0,1),constant_values=np.nan)
     back_fwd_dilation                           = np.vstack([back_dilation,fwd_dilation])
 
-    max_dilation_speed                          = np.empty_like(dia)
+    max_dilation_speed                          = np.full_like(dia,np.nan)
     max_dilation_speed[is_valid]                = np.nanmax(np.abs(back_fwd_dilation),axis=0)
     max_dilation_speed
 
@@ -346,4 +346,4 @@ def process_run(raw_fname):
     args = parser.parse_args()
 
     raw_fname = args.fname
-    process_run(raw_fname)
+    process_run(raw_fname)