From dd21f9440fa03e89c5bc98d7d6c588607958f7c5 Mon Sep 17 00:00:00 2001
From: Sunil Samra <Sunil.Samra@nsc.co.uk>
Date: Mon, 17 Jul 2023 14:21:37 +0000
Subject: [PATCH] Apply suggestions from code review

---
 src/primaite/environment/observations.py | 2 --
 1 file changed, 2 deletions(-)

diff --git a/src/primaite/environment/observations.py b/src/primaite/environment/observations.py
index 70f3cdde..b517679c 100644
--- a/src/primaite/environment/observations.py
+++ b/src/primaite/environment/observations.py
@@ -702,8 +702,6 @@ class ObservationsHandler:
         # Instantiate the handler
         handler = cls()
 
-        if obs_space_config.get("flatten"):
-            handler.flatten = True
 
         for component_cfg in obs_space_config["components"]:
             # Figure out which class can instantiate the desired component