Merge pull request #1747 from colleenjg/GH-1741/bugfix/running-speed-…

…main Fixes 2 errors in allensdk.brain_observatory.extract_running_speed.__…
AllenInstitute · Oct 15, 2020 · 18f49c7 · 18f49c7
2 parents 7dd9dfe + 9c07065
commit 18f49c7
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/allensdk/brain_observatory/extract_running_speed/__main__.py b/allensdk/brain_observatory/extract_running_speed/__main__.py
@@ -60,9 +60,9 @@ def extract_running_speeds(
 
     durations = end_times - start_times
     if use_median_duration:
-        angular_velocity = dx_rad / durations
+        angular_velocity = dx_rad / np.median(durations)
     else:
-        angular_velocity = dx_rad = np.median(durations)
+        angular_velocity = dx_rad / durations
 
     radius = wheel_radius * subject_position
     linear_velocity = angular_to_linear_velocity(angular_velocity, radius)