From 40da84393c7a82b0e2ba9554bf2c50c832dab747 Mon Sep 17 00:00:00 2001 From: ebuehrle <43623224+ebuehrle@users.noreply.github.com> Date: Tue, 14 Sep 2021 18:02:56 +0200 Subject: [PATCH] Update `imitation` version --- scratch/etienne/intersimple/data/expert.py | 8 +++++++- 1 file changed, 7 insertions(+), 1 deletion(-) diff --git a/scratch/etienne/intersimple/data/expert.py b/scratch/etienne/intersimple/data/expert.py index 86ce357..a8edbaf 100644 --- a/scratch/etienne/intersimple/data/expert.py +++ b/scratch/etienne/intersimple/data/expert.py @@ -63,10 +63,16 @@ class NormalizedIntersimpleExpert(IntersimpleExpert): action, _ = super().predict(*args, **kwargs) return self._intersimple._normalize(action), None -class DummyVecEnvPolicy(): +class DummyVecEnvPolicy(BasePolicy): def __init__(self, experts): self._experts = [e() for e in experts] + + def forward(self, *args, **kwargs): + raise NotImplementedError() + + def _predict(self, *args, **kwargs): + raise NotImplementedError() def predict(self, *args, **kwargs): predictions = [e.predict() for e in self._experts]