add requirements fix

Camyll · Camyll · commit 7beaa914fd13 · 2025-04-10T15:43:30.000-07:00
diff --git 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/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/requirements.txt b/torchbenchmark/models/soft_actor_critic/requirements.txt
@@ -1,3 +1,3 @@
-gym
+gymnasium
 pygame
 tensorboardX

-Original file line number
+Diff line change
@@ @@ -1,3 +1,3 @@ @@
 kornia
 scikit-image
 -gym
 +gymnasium