Skip to content

Commit b1917f1

Browse files
authored
[Benchmark] Fix benchmark CI by adding missing deps (#3163)
1 parent d58723e commit b1917f1

File tree

2 files changed

+2
-2
lines changed

2 files changed

+2
-2
lines changed

.github/workflows/benchmarks.yml

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -81,7 +81,7 @@ jobs:
8181
${{ matrix.device == 'CPU' && 'export CUDA_VISIBLE_DEVICES=' || '' }}
8282
8383
python3.10 -m pip install --pre torch torchvision --index-url https://download.pytorch.org/whl/nightly/cu128 -U
84-
python3.10 -m pip install ninja pytest pytest-benchmark mujoco dm_control "gym[accept-rom-license,atari]"
84+
python3.10 -m pip install ninja pytest pytest-benchmark mujoco dm_control "gym[accept-rom-license,atari]" transformers
8585
python3 -m pip install "pybind11[global]"
8686
python3.10 -m pip install git+https://github.com/pytorch/tensordict
8787
python3.10 -m pip install safetensors tqdm pandas numpy matplotlib

.github/workflows/benchmarks_pr.yml

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -79,7 +79,7 @@ jobs:
7979
${{ matrix.device == 'CPU' && 'export CUDA_VISIBLE_DEVICES=' || '' }}
8080
8181
python3.10 -m pip install --pre torch torchvision --index-url https://download.pytorch.org/whl/nightly/cu128 -U
82-
python3.10 -m pip install ninja pytest pytest-benchmark mujoco dm_control "gym[accept-rom-license,atari]"
82+
python3.10 -m pip install ninja pytest pytest-benchmark mujoco dm_control "gym[accept-rom-license,atari]" transformers
8383
python3.10 -m pip install "pybind11[global]"
8484
python3.10 -m pip install git+https://github.com/pytorch/tensordict
8585
python3.10 -m pip install safetensors tqdm pandas numpy matplotlib

0 commit comments

Comments
 (0)