Fix so that flatten works

bpiwowar · Nov 20, 2023 · 60a8c81 · 60a8c81
1 parent ae9506d
commit 60a8c81
Show file tree

Hide file tree

Showing 2 changed files with 3 additions and 2 deletions.
diff --git a/src/pystk2_gymnasium/envs.py b/src/pystk2_gymnasium/envs.py
@@ -188,7 +188,7 @@ def reset_race(
         self.current_track = self.default_track
         if self.current_track is None:
             self.current_track = self.TRACKS[random.randint(0, len(self.TRACKS))]
-            logging.info("Selected %s", self.current_track)
+            logging.debug("Selected %s", self.current_track)
         self.config = pystk2.RaceConfig(
             num_kart=self.num_kart,
             seed=random.randint(2**16),

diff --git a/src/pystk2_gymnasium/wrappers.py b/src/pystk2_gymnasium/wrappers.py
@@ -31,7 +31,8 @@ def __init__(self, space: gym.Space):
         highs = []
         counts = []
 
-        for key, value in space.items():
+        # Combine keys (sort them before hand)
+        for key, value in sorted(space.items(), key=lambda x: x[0]):
             # Ignore the AI action
             if key == "action":
                 continue