GymEnv utility methods

osigaud · Jun 2, 2023 · fa8597a · fa8597a
1 parent 1e62120
commit fa8597a
Showing 1 changed file with 12 additions and 0 deletions.
diff --git a/src/bbrl/agents/gymnasium.py b/src/bbrl/agents/gymnasium.py
@@ -200,6 +200,18 @@ def get_obs_and_actions_sizes(self):
             state_dim = 1  # self.observation_space.n
         return state_dim, action_dim
 
+    def is_continuous_action(self):
+        return isinstance(self.action_space, gym.spaces.Box)
+
+    def is_discrete_action(self):
+        return isinstance(self.action_space, gym.spaces.Discrete)
+
+    def is_continuous_state(self):
+        return isinstance(self.observation_space, gym.spaces.Box)
+
+    def is_discrete_state(self):
+        return isinstance(self.observation_space, gym.spaces.Discrete)
+
 class ParallelGymAgent(GymAgent):
     """Create an Agent from a gymnasium environment