diff --git a/miniworld/manual_control.py b/miniworld/manual_control.py index a0e69a1c..bb7c12a1 100644 --- a/miniworld/manual_control.py +++ b/miniworld/manual_control.py @@ -6,7 +6,7 @@ class ManualControl: def __init__(self, env, no_time_limit, domain_rand): - self.env = env + self.env = env.unwrapped if no_time_limit: self.env.max_episode_steps = math.inf