deprecate gym to gymnasium and add timm yaml file back

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

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

torchbenchmark/models/drq/requirements.txt

@@ -1,3 +1,3 @@
 kornia
 scikit-image
-gym
+gymnasium

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,

torchbenchmark/models/soft_actor_critic/requirements.txt

@@ -1,3 +1,3 @@
-gym
+gymnasium
 pygame
 tensorboardX

userbenchmark/dynamo/dynamobench/timm_models.yaml

@@ -0,0 +1,4 @@
+# removing this file prevents the TimmRunner from dynamically skipping models
+skip:
+  all:
+    - ~