Adapt expert for RandomLocations environment

2021-10-27 13:56:20 +02:00
parent 97b98cb52e
commit 039f8a7643
2 changed files with 5 additions and 1 deletions
--- a/scratch/etienne/intersimple/data/expert.py
+++ b/scratch/etienne/intersimple/data/expert.py
@@ -49,6 +49,8 @@ class IntersimpleExpert(BasePolicy):
        raise NotImplementedError()

    def _action(self):
+        # RandomLocation mixin re-initializes the intersim sub-env
+        self._intersim_expert._intersim = self._intersimple._env
        return self._intersim_expert._action()[self._intersimple._agent]

    def predict(self, *args, **kwargs):