type hinting again

reginald-mclean · Aug 30, 2024 · 99ba23a · 99ba23a
1 parent 19f2062
commit 99ba23a
Showing 1 changed file with 5 additions and 5 deletions.
diff --git a/metaworld/wrappers.py b/metaworld/wrappers.py
@@ -1,4 +1,4 @@
-from typing import Any, List, Optional
+from typing import Any, Dict, List, Optional
 
 import gymnasium as gym
 import numpy as np
@@ -60,14 +60,14 @@ def toggle_sample_tasks_on_reset(self, on: bool):
         self.sample_tasks_on_reset = on
 
     def reset(
-        self, *, seed: Optional[int] = None, options: Optional[dict[str, Any]] = None
+        self, *, seed: Optional[int] = None, options: Optional[Dict[str, Any]] = None
     ):
         if self.sample_tasks_on_reset:
             self._set_random_task()
         return self.env.reset(seed=seed, options=options)
 
     def sample_tasks(
-        self, *, seed: Optional[int] = None, options: Optional[dict[str, Any]] = None
+        self, *, seed: Optional[int] = None, options: Optional[Dict[str, Any]] = None
     ):
         self._set_random_task()
         return self.env.reset(seed=seed, options=options)
@@ -110,14 +110,14 @@ def __init__(
             np.random.seed(seed)
 
     def reset(
-        self, *, seed: Optional[int] = None, options: Optional[dict[str, Any]] = None
+        self, *, seed: Optional[int] = None, options: Optional[Dict[str, Any]] = None
     ):
         if self.sample_tasks_on_reset:
             self._set_pseudo_random_task()
         return self.env.reset(seed=seed, options=options)
 
     def sample_tasks(
-        self, *, seed: Optional[int] = None, options: Optional[dict[str, Any]] = None
+        self, *, seed: Optional[int] = None, options: Optional[Dict[str, Any]] = None
     ):
         self._set_pseudo_random_task()
         return self.env.reset(seed=seed, options=options)