Merge branch 'main' of github.com:sisl/InteractionImitation

2021-10-28 09:45:27 +02:00
parent d89c77409c 039f8a7643
commit 214c6daeaa
3 changed files with 96 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):