Fix action space order

edbeeching · Dec 3, 2023 · c1dab7e · c1dab7e
1 parent ac7241b
commit c1dab7e
Showing 1 changed file with 2 additions and 1 deletion.
diff --git a/godot_rl/core/godot_env.py b/godot_rl/core/godot_env.py
@@ -12,6 +12,7 @@
 from typing import Optional
 from godot_rl.core.utils import ActionSpaceProcessor, convert_macos_path
 
+from collections import OrderedDict
 
 class GodotEnv:
     MAJOR_VERSION = "0" # Versioning for the environment
@@ -319,7 +320,7 @@ def _get_env_info(self):
 
         # actions can be "single" for a single action head
         # or "multi" for several outputeads
-        action_spaces = {}
+        action_spaces = OrderedDict()
         print("action space", json_dict["action_space"])
         for k, v in json_dict["action_space"].items():
             if v["action_type"] == "discrete":