diff --git a/metaworld/wrappers.py b/metaworld/wrappers.py index 092da2941..f14b8bce9 100644 --- a/metaworld/wrappers.py +++ b/metaworld/wrappers.py @@ -98,7 +98,7 @@ def toggle_sample_tasks_on_reset(self, on: bool): def __init__( self, env: Env, - tasks: list[Task], + tasks: List[Task], sample_tasks_on_reset: bool = False, seed: Optional[int] = None, ):