forward args and kwargs in VAEWrappedEnvPathCollector

KunBB · Apr 11, 2019 · 1ac7b1e · 1ac7b1e
1 parent f91f72c
commit 1ac7b1e
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/rlkit/samplers/data_collector.py b/rlkit/samplers/data_collector.py
@@ -198,7 +198,7 @@ def __init__(
         self._goal_sampling_mode = goal_sampling_mode
         self._decode_goals = decode_goals
 
-    def collect_new_paths(self, *args):
+    def collect_new_paths(self, *args, **kwargs):
         self._env.goal_sampling_mode = self._goal_sampling_mode
         self._env.decode_goals = self._decode_goals
-        return super().collect_new_paths(*args)
+        return super().collect_new_paths(*args, **kwargs)