gymnasium -> gym

pytorch · Nov 20, 2023 · 23a4786 · 23a4786
1 parent 6189be5
commit 23a4786
Show file tree

Hide file tree

Showing 2 changed files with 4 additions and 4 deletions.
diff --git a/.github/workflows/benchmarks.yml b/.github/workflows/benchmarks.yml
@@ -34,7 +34,7 @@ jobs:
           python -m pip install git+https://github.com/pytorch/tensordict 
           python setup.py develop
           python -m pip install pytest pytest-benchmark
-          python3 -m pip install "gymnasium[accept-rom-license,atari]"
+          python3 -m pip install "gym[accept-rom-license,atari]"
           python3 -m pip install dm_control
       - name: Run benchmarks
         run: |
@@ -95,7 +95,7 @@ jobs:
           python3 -m pip install git+https://github.com/pytorch/tensordict 
           python3 setup.py develop
           python3 -m pip install pytest pytest-benchmark
-          python3 -m pip install "gymnasium[accept-rom-license,atari]"
+          python3 -m pip install "gym[accept-rom-license,atari]"
           python3 -m pip install dm_control
       - name: check GPU presence
         run: |

diff --git a/.github/workflows/benchmarks_pr.yml b/.github/workflows/benchmarks_pr.yml
@@ -33,7 +33,7 @@ jobs:
           python -m pip install git+https://github.com/pytorch/tensordict 
           python setup.py develop
           python -m pip install pytest pytest-benchmark
-          python3 -m pip install "gymnasium[accept-rom-license,atari]"
+          python3 -m pip install "gym[accept-rom-license,atari]"
           python3 -m pip install dm_control
       - name: Setup benchmarks
         run: |
@@ -106,7 +106,7 @@ jobs:
           python3 -m pip install git+https://github.com/pytorch/tensordict 
           python3 setup.py develop
           python3 -m pip install pytest pytest-benchmark
-          python3 -m pip install "gymnasium[accept-rom-license,atari]"
+          python3 -m pip install "gym[accept-rom-license,atari]"
           python3 -m pip install dm_control
       - name: check GPU presence
         run: |