Fix: StateObservable sometimes having wrong low-bound

2022-12-13 19:50:57 +01:00 · 2022-12-13 19:50:57 +01:00 · 35864d4b38
commit 35864d4b38
parent f421c92f83
1 changed files with 1 additions and 1 deletions
--- a/columbus/observables.py
+++ b/columbus/observables.py
@ -250,7 +250,7 @@ class StateObservable(Observable):
        self.reset()
        num = len(self.entities)*2+len(self._timeoutEntities) + \
            self.speedAgent*2 + self.include_rand
-        return spaces.Box(low=0-1*self.coordsRelativeToAgent, high=1,
+        return spaces.Box(low=0-1*(self.coordsRelativeToAgent or self.speedAgent), high=1,
                          shape=(num,), dtype=np.float64)

    def get_observation(self):