From e9a9f05f2379bceb5e8c76e32cd1aac3388bb28d Mon Sep 17 00:00:00 2001
From: robfiras <f.al-hafez@posteo.de>
Date: Wed, 11 Sep 2024 17:48:30 +0200
Subject: [PATCH] fixed small bug in data generation pipeline.

---
 loco_mujoco/utils/dataset.py | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/loco_mujoco/utils/dataset.py b/loco_mujoco/utils/dataset.py
index 41b7ad6..eb820aa 100644
--- 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