From 7e6fe2759b02694b3d19d04cff7b94f5be86ef22 Mon Sep 17 00:00:00 2001
From: Marek Wolan <marek.wolan@methods.co.uk>
Date: Fri, 30 Jun 2023 15:43:15 +0100
Subject: [PATCH] Fix flattening when there are no components.

---
 src/primaite/environment/observations.py | 5 ++++-
 1 file changed, 4 insertions(+), 1 deletion(-)

diff --git a/src/primaite/environment/observations.py b/src/primaite/environment/observations.py
index fcd52559..b19bd29f 100644
--- a/src/primaite/environment/observations.py
+++ b/src/primaite/environment/observations.py
@@ -450,7 +450,10 @@ class ObservationsHandler:
             self._space = component_spaces[0]
         else:
             self._space = spaces.Tuple(component_spaces)
-        self._flat_space = spaces.flatten_space(self._space)
+        if len(component_spaces) > 0:
+            self._flat_space = spaces.flatten_space(self._space)
+        else:
+            self._flat_space = spaces.Box(0, 1, (0,))
 
     @property
     def space(self):