Merge branch 'dev' into mc24_p2

MyoHub · Sep 18, 2024 · e26326a · e26326a
2 parents 8cdfe5b + f664496
commit e26326a
Showing 1 changed file with 3 additions and 0 deletions.
diff --git a/myosuite/envs/myo/myochallenge/bimanual_v0.py b/myosuite/envs/myo/myochallenge/bimanual_v0.py
@@ -257,6 +257,9 @@ def get_reward_dict(self, obs_dict):
         elbow_err = 5 * np.exp(-10 * (obs_dict['elbow_fle'][0] - 1.) ** 2) - 5
         goal_dis = np.array(
             [[np.abs(np.linalg.norm(obj_pos[:2] - goal_pos, axis=-1))]])
+
+        isTimeLimit = obs_dict['time'] > 3.0
+
         rwd_dict = collections.OrderedDict(
             (
                 # Optional Keys