add termination only for chase task

MyoHub · Oct 5, 2023 · 37e6f8e · 37e6f8e
1 parent c3f12cb
commit 37e6f8e
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/myosuite/envs/myo/myochallenge/chasetag_v0.py b/myosuite/envs/myo/myochallenge/chasetag_v0.py
@@ -627,7 +627,7 @@ def _win_condition(self):
 
     def _lose_condition(self):
         # falling on knees is always termination
-        if self._get_fallen_condition():
+        if self._get_fallen_condition() and self.current_task == 'chase':
             return 1
         if self.current_task == 'chase':
             return self._chase_lose_condition()