set vel to nan rather than inf

Author: ecobost

src/spikeinterface/metrics/template/metrics.py

@@ -361,8 +361,8 @@ def get_velocity_fits(template, channel_locations, sampling_frequency, **kwargs)
         peak_times_ms_above = np.argmin(template_above, 0) / sampling_frequency * 1000 - max_peak_time
         distances_um_above = np.array([np.linalg.norm(cl - max_channel_location) for cl in channel_locations_above])
         inv_velocity_above, score = fit_line_robust(distances_um_above, peak_times_ms_above)
-        if score > min_r2:
-            velocity_above = 1 / inv_velocity_above if np.abs(inv_velocity_above) > 1e-9 else np.inf
+        if score > min_r2 and np.abs(inv_velocity_above) > 1e-9:
+            velocity_above = 1 / inv_velocity_above
         else:
             velocity_above = np.nan
 
@@ -376,8 +376,8 @@ def get_velocity_fits(template, channel_locations, sampling_frequency, **kwargs)
         peak_times_ms_below = np.argmin(template_below, 0) / sampling_frequency * 1000 - max_peak_time
         distances_um_below = np.array([np.linalg.norm(cl - max_channel_location) for cl in channel_locations_below])
         inv_velocity_below, score = fit_line_robust(distances_um_below, peak_times_ms_below)
-        if score > min_r2:
-            velocity_below = 1 / inv_velocity_below if np.abs(inv_velocity_below) > 1e-9 else np.inf
+        if score > min_r2 and np.abs(inv_velocity_below) > 1e-9:
+            velocity_below = 1 / inv_velocity_below
         else:
             velocity_below = np.nan