Upgrade Gymanasium to v1.0.0

takuseno · Oct 12, 2024 · b699b52 · b699b52
1 parent a096424
commit b699b52
Show file tree

Hide file tree

Showing 4 changed files with 6 additions and 6 deletions.
diff --git a/d3rlpy/datasets.py b/d3rlpy/datasets.py
@@ -13,7 +13,7 @@
 from gym.wrappers.time_limit import TimeLimit
 from gymnasium.spaces import Box as GymnasiumBox
 from gymnasium.spaces import Dict as GymnasiumDictSpace
-from gymnasium.wrappers.time_limit import TimeLimit as GymnasiumTimeLimit
+from gymnasium.wrappers import TimeLimit as GymnasiumTimeLimit
 
 from .dataset import (
     BasicTrajectorySlicer,

diff --git a/examples/deepmind_control.py b/examples/deepmind_control.py
@@ -1,7 +1,7 @@
 import argparse
 
 import gymnasium
-from gymnasium.wrappers.flatten_observation import FlattenObservation
+from gymnasium.wrappers import FlattenObservation
 
 import d3rlpy
 
@@ -21,8 +21,8 @@ def main() -> None:
     args = parser.parse_args()
 
     env_id = f"dm_control/{args.env}"
-    env = FlattenObservation(gymnasium.make(env_id))
-    eval_env = FlattenObservation(gymnasium.make(env_id))
+    env = FlattenObservation(gymnasium.make(env_id))  # type: ignore
+    eval_env = FlattenObservation(gymnasium.make(env_id))  # type: ignore
 
     # fix seed
     d3rlpy.seed(args.seed)

diff --git a/requirements.txt b/requirements.txt
@@ -6,4 +6,4 @@ click==8.0.1
 typing-extensions==3.7.4.3
 structlog==20.2.0
 colorama==0.4.4
-gymnasium==0.29.0
+gymnasium==1.0.0
diff --git a/setup.py b/setup.py
@@ -42,7 +42,7 @@
             "structlog",
             "colorama",
             "dataclasses-json",
-            "gymnasium",
+            "gymnasium>=1.0.0",
         ],
         packages=find_packages(exclude=["tests*"]),
         python_requires=">=3.8.0",