Skip to content

Commit f26ad38

Browse files
committed
test if this helps
1 parent 6eb17f1 commit f26ad38

File tree

3 files changed

+7
-3
lines changed

3 files changed

+7
-3
lines changed

torchbenchmark/models/drq/drqutils.py

+1-1
Original file line numberDiff line numberDiff line change
@@ -3,7 +3,7 @@
33
import random
44
from collections import deque
55

6-
import gym
6+
import gymnasium as gym
77

88
import numpy as np
99
import torch

torchbenchmark/models/soft_actor_critic/envs.py

+2-2
Original file line numberDiff line numberDiff line change
@@ -2,7 +2,7 @@
22
import random
33
from collections import deque
44

5-
import gym
5+
import gymnasium as gym
66
import numpy as np
77

88

@@ -132,7 +132,7 @@ def highway_env(env_id):
132132
133133
highway_env: https://highway-env.readthedocs.io/en/latest/index.html
134134
"""
135-
import gym
135+
import gymnasium as gym
136136

137137
env = gym.make(env_id)
138138
env.configure({"action": {"type": "ContinuousAction"}})
Original file line numberDiff line numberDiff line change
@@ -0,0 +1,4 @@
1+
# removing this file prevents the TimmRunner from dynamically skipping models
2+
skip:
3+
all:
4+
- ~

0 commit comments

Comments
 (0)