fixed small bug in data generation pipeline.

robfiras · Sep 11, 2024 · e9a9f05 · e9a9f05
1 parent 7ecafe2
commit e9a9f05
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/loco_mujoco/utils/dataset.py b/loco_mujoco/utils/dataset.py
@@ -197,7 +197,7 @@ def adapt_mocap(path, joint_conf, unavailable_keys, rename_map=None, discard_fir
     # if needed discard first and last part of the dataset
     for j_name, val in dataset.items():
         val_temp = val[discard_first:]
-        val_temp = val_temp[0:-discard_last]
+        val_temp = val_temp[0:-discard_last] if discard_last > 0 else val_temp
         dataset[j_name] = val_temp
 
     return dataset