diff --git a/torchbenchmark/models/drq/__init__.py b/torchbenchmark/models/drq/__init__.py index 63da8ef154..351e5e9a11 100644 --- a/torchbenchmark/models/drq/__init__.py +++ b/torchbenchmark/models/drq/__init__.py @@ -4,7 +4,7 @@ import numpy as np import torch -from gym import spaces +from gymnasium import spaces from torchbenchmark.tasks import REINFORCEMENT_LEARNING from ...util.model import BenchmarkModel diff --git a/torchbenchmark/models/drq/drqutils.py b/torchbenchmark/models/drq/drqutils.py index 24be20f825..53c0a1da8b 100644 --- a/torchbenchmark/models/drq/drqutils.py +++ b/torchbenchmark/models/drq/drqutils.py @@ -3,7 +3,7 @@ import random from collections import deque -import gym +import gymnasium as gym import numpy as np import torch diff --git a/torchbenchmark/models/drq/requirements.txt b/torchbenchmark/models/drq/requirements.txt index cfbd5fcdd6..18f7d23761 100644 --- a/torchbenchmark/models/drq/requirements.txt +++ b/torchbenchmark/models/drq/requirements.txt @@ -1,3 +1,3 @@ kornia scikit-image -gym +gymnasium diff --git a/torchbenchmark/models/soft_actor_critic/envs.py b/torchbenchmark/models/soft_actor_critic/envs.py index 7e39bba8ec..07ae22151b 100644 --- a/torchbenchmark/models/soft_actor_critic/envs.py +++ b/torchbenchmark/models/soft_actor_critic/envs.py @@ -2,7 +2,7 @@ import random from collections import deque -import gym +import gymnasium as gym import numpy as np @@ -132,7 +132,7 @@ def highway_env(env_id): highway_env: https://highway-env.readthedocs.io/en/latest/index.html """ - import gym + import gymnasium as gym env = gym.make(env_id) env.configure({"action": {"type": "ContinuousAction"}}) @@ -339,15 +339,15 @@ def load_dmc( """ Load a task from the deepmind control suite. - Uses dmc2gym (https://github.com/denisyarats/dmc2gym) + Uses dmc2gymnasiym (https://github.com/imgeorgiev/dmc2gymnasium) Note that setting seed=None (the default) picks a random seed """ - import dmc2gym + import dmc2gymnasium if seed is None: seed = random.randint(1, 100000) - env = dmc2gym.make( + env = dmc2gymnasium.make( domain_name=domain_name, task_name=task_name, from_pixels=from_pixels, diff --git a/torchbenchmark/models/soft_actor_critic/requirements.txt b/torchbenchmark/models/soft_actor_critic/requirements.txt index 9dd7425d1e..6829437f67 100644 --- a/torchbenchmark/models/soft_actor_critic/requirements.txt +++ b/torchbenchmark/models/soft_actor_critic/requirements.txt @@ -1,3 +1,3 @@ -gym +gymnasium pygame tensorboardX diff --git a/userbenchmark/dynamo/dynamobench/timm_models.yaml b/userbenchmark/dynamo/dynamobench/timm_models.yaml new file mode 100644 index 0000000000..8fe214c8b6 --- /dev/null +++ b/userbenchmark/dynamo/dynamobench/timm_models.yaml @@ -0,0 +1,4 @@ +# removing this file prevents the TimmRunner from dynamically skipping models +skip: + all: + - ~