Skip to content

Commit a66d131

Browse files
authored
[TPU][Bugfix][CI] Fix broken tests/build dependency (#25255)
Signed-off-by: NickLucche <[email protected]>
1 parent 21467f9 commit a66d131

File tree

2 files changed

+2
-2
lines changed

2 files changed

+2
-2
lines changed

.buildkite/scripts/hardware_ci/run-tpu-v1-test-part2.sh

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -62,7 +62,7 @@ echo "--- Installing Python dependencies ---"
6262
python3 -m pip install --progress-bar off git+https://github.com/thuml/depyf.git \
6363
&& python3 -m pip install --progress-bar off pytest pytest-asyncio tpu-info \
6464
&& python3 -m pip install --progress-bar off "lm-eval @ git+https://github.com/EleutherAI/lm-evaluation-harness.git@206b7722158f58c35b7ffcd53b035fdbdda5126d" \
65-
&& python3 -m pip install --progress-bar off hf-transfer
65+
&& python3 -m pip install --progress-bar off hf-transfer tblib==3.1.0
6666
echo "--- Python dependencies installed ---"
6767
export VLLM_USE_V1=1
6868
export VLLM_XLA_CHECK_RECOMPILATION=1

.buildkite/scripts/hardware_ci/run-tpu-v1-test.sh

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -62,7 +62,7 @@ echo "--- Installing Python dependencies ---"
6262
python3 -m pip install --progress-bar off git+https://github.com/thuml/depyf.git \
6363
&& python3 -m pip install --progress-bar off pytest pytest-asyncio tpu-info \
6464
&& python3 -m pip install --progress-bar off "lm-eval @ git+https://github.com/EleutherAI/lm-evaluation-harness.git@206b7722158f58c35b7ffcd53b035fdbdda5126d" \
65-
&& python3 -m pip install --progress-bar off hf-transfer
65+
&& python3 -m pip install --progress-bar off hf-transfer tblib==3.1.0
6666
echo "--- Python dependencies installed ---"
6767
export VLLM_USE_V1=1
6868
export VLLM_XLA_CHECK_RECOMPILATION=1

0 commit comments

Comments
 (0)