Fixing an OOB during the classification pass if you have no force data

keenon · Sep 28, 2024 · 90f8c52 · 90f8c52
1 parent 7958c21
commit 90f8c52
Showing 1 changed file with 2 additions and 0 deletions.
diff --git a/server/engine/src/dynamics_pass/classification_pass.py b/server/engine/src/dynamics_pass/classification_pass.py
@@ -5,6 +5,8 @@
 
 def get_num_steps(raw_force_plate_forces: List[List[np.ndarray]],
                   raw_force_plate_cops: List[List[np.ndarray]]) -> Tuple[int, List[int]]:
+    if len(raw_force_plate_forces) == 0:
+        return 0, []
     num_force_plates = len(raw_force_plate_forces)
     trial_len = len(raw_force_plate_forces[0])
     num_steps = 0