Fixed all gymnasium warnings

[ci skip]
bpiwowar · Nov 10, 2023 · 9d3f3bd · 9d3f3bd
1 parent 442966f
commit 9d3f3bd
Show file tree

Hide file tree

Showing 2 changed files with 5 additions and 3 deletions.
diff --git a/src/pystk2_gymnasium/envs.py b/src/pystk2_gymnasium/envs.py
@@ -332,13 +332,13 @@ def sort_closest(positions, *lists):
             action = self.race.get_kart_action(kart_ix)
             obs = {
                 "action": {
-                    "steer": action.steer,
                     "brake": action.brake,
                     "nitro": action.nitro,
                     "drift": action.drift,
                     "rescue": action.rescue,
                     "fire": action.fire,
-                    "acceleration": action.acceleration,
+                    "steer": np.array([action.steer], dtype=np.float32),
+                    "acceleration": np.array([action.acceleration], dtype=np.float32),
                 }
             }
 

diff --git a/tests/test_envs.py b/tests/test_envs.py
@@ -21,12 +21,14 @@ def test_env(name, use_ai):
         done = False
         state, *_ = env.reset()
 
-        while not done:
+        for _ in range(10):
             ix += 1
             action = env.action_space.sample()
             # print(action)
             state, reward, terminated, truncated, _ = env.step(action)
             done = truncated or terminated
+            if done: 
+                break
     finally:
         if env is not None:
             env.close()