[python/tests/utils]game_controller.py] Propagate exceptions (#244)

google-deepmind · Jan 4, 2023 · b1db91a · b1db91a
1 parent ea4219e
commit b1db91a
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/python/tests/utils/game_controller.py b/python/tests/utils/game_controller.py
@@ -218,11 +218,11 @@ def rotation_velocity(self):
   def _observations(self):
     try:
       return self._env.observations()
-    except RuntimeError:
+    except RuntimeError as e:
       if self._env.is_running():
         raise
       else:
-        raise EpisodeFinishedError()
+        raise EpisodeFinishedError from e
 
   def _get_empty_actions(self):
     return np.zeros([len(self._action_index)], dtype=np.intc)