Fix flattening when there are no components.

2023-06-30 15:43:15 +01:00
parent d86489a9c2
commit 7e6fe2759b
1 changed files with 4 additions and 1 deletions
--- 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):