Fixed bugged compass Observable

2022-08-27 21:07:31 +02:00 · 2022-08-27 21:07:31 +02:00 · 0a458b863d
commit 0a458b863d
parent 6425a4f829
1 changed files with 4 additions and 1 deletions
--- a/columbus/observables.py
+++ b/columbus/observables.py
@ -301,11 +301,14 @@ class CompassObservable(Observable):
        return self._entities

    def get_observation_space(self):
-        self.env.reset()
+        self.reset()
        num = len(self.entities)*2
        return spaces.Box(low=-1, high=1,
                          shape=(num,), dtype=np.float32)

+    def reset(self):
+        self._entities = None
+
    def get_observation(self):
        obs = []
        for entity in self.entities: