From 65f2d6202f0f6dc70d4d7c1d62342fd5ed964ddb Mon Sep 17 00:00:00 2001
From: Marek Wolan <marek.wolan@methods.co.uk>
Date: Wed, 31 May 2023 10:51:29 +0100
Subject: [PATCH] Add default observation type

---
 src/primaite/environment/primaite_env.py | 4 ++--
 1 file changed, 2 insertions(+), 2 deletions(-)

diff --git a/src/primaite/environment/primaite_env.py b/src/primaite/environment/primaite_env.py
index 7f102bd4..f57b274d 100644
--- 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: