Add kwargs in reset. (#695)

Author: ernestum

src/imitation/data/wrappers.py

@@ -186,7 +186,7 @@ def __init__(self, env: gym.Env):
         self._rews = None
 
     def reset(self, **kwargs):
-        new_obs = super().reset()
+        new_obs = super().reset(**kwargs)
         self._obs = [new_obs]
         self._rews = []
         return new_obs