Add default observation type

2023-05-31 10:51:29 +01:00
parent 733025bd53
commit 65f2d6202f
1 changed files with 2 additions and 2 deletions
--- a/src/primaite/environment/primaite_env.py
+++ b/src/primaite/environment/primaite_env.py
@@ -149,8 +149,8 @@ class Primaite(Env):
        # The action type
        self.action_type = 0

-        # Observation type.
-        self.observation_type = 0
+        # Observation type, by default box.
+        self.observation_type = ObservationType.BOX

        # Open the config file and build the environment laydown
        try: