Merge pull request #101 from elladyr/dev

Merge for P2
MyoHub · Oct 6, 2023 · 24cd961 · 24cd961
2 parents f389938 + cdcf556
commit 24cd961
Showing 1 changed file with 2 additions and 1 deletion.
diff --git a/myosuite/envs/myo/myochallenge/chasetag_v0.py b/myosuite/envs/myo/myochallenge/chasetag_v0.py
@@ -387,6 +387,7 @@ def _setup(self,
                **kwargs,
                ):
 
+
         self._setup_convenience_vars()
         # check that this works everywhere and is efficient.
         self.heightfield = HeightField(sim=self.sim,
@@ -459,7 +460,7 @@ def get_reward_dict(self, obs_dict):
             score = self._get_score(float(self.obs_dict['time'])) if win_cdt else 0
             self.obs_dict['time'] = self.maxTime if lose_cdt else self.obs_dict['time']
         elif self.current_task == 'evade':
-            score = self._get_score(float(self.obs_dict['time']))
+            score = self._get_score(float(self.obs_dict['time'])) if (win_cdt or lose_cdt) else 0
         # ----------------------
 
         # Example reward, you should change this!