Skip to content

Commit 839ab00

Browse files
authored
Re-enable Xet on TPU tests now that hf_xet has been updated (#22666)
Signed-off-by: Harry Mellor <[email protected]>
1 parent 9b94d6e commit 839ab00

File tree

3 files changed

+2
-5
lines changed

3 files changed

+2
-5
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
@@ -128,7 +128,7 @@ run_and_track_test() {
128128
129129
# --- Actual Test Execution ---
130130
run_and_track_test 1 "test_struct_output_generate.py" \
131-
"HF_HUB_DISABLE_XET=1 python3 -m pytest -s -v /workspace/vllm/tests/v1/entrypoints/llm/test_struct_output_generate.py -k \"not test_structured_output_with_reasoning_matrices\""
131+
"python3 -m pytest -s -v /workspace/vllm/tests/v1/entrypoints/llm/test_struct_output_generate.py -k \"not test_structured_output_with_reasoning_matrices\""
132132
run_and_track_test 2 "test_moe_pallas.py" \
133133
"python3 -m pytest -s -v /workspace/vllm/tests/tpu/test_moe_pallas.py"
134134
run_and_track_test 3 "test_lora.py" \

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

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -134,7 +134,7 @@ run_and_track_test 1 "test_compilation.py" \
134134
run_and_track_test 2 "test_basic.py" \
135135
"python3 -m pytest -s -v /workspace/vllm/tests/v1/tpu/test_basic.py"
136136
run_and_track_test 3 "test_accuracy.py::test_lm_eval_accuracy_v1_engine" \
137-
"HF_HUB_DISABLE_XET=1 python3 -m pytest -s -v /workspace/vllm/tests/entrypoints/llm/test_accuracy.py::test_lm_eval_accuracy_v1_engine"
137+
"python3 -m pytest -s -v /workspace/vllm/tests/entrypoints/llm/test_accuracy.py::test_lm_eval_accuracy_v1_engine"
138138
run_and_track_test 4 "test_quantization_accuracy.py" \
139139
"python3 -m pytest -s -v /workspace/vllm/tests/tpu/test_quantization_accuracy.py"
140140
run_and_track_test 5 "examples/offline_inference/tpu.py" \

tests/entrypoints/llm/test_accuracy.py

Lines changed: 0 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -96,9 +96,6 @@ def test_lm_eval_accuracy_v1_engine_fp8_kv_cache(
9696
more_args = None
9797
if current_platform.is_tpu():
9898
# Limit compilation time for TPU V1
99-
100-
# xet doesn't work well for Qwen/Qwen3-1.7B
101-
m.setenv("HF_HUB_DISABLE_XET", "1")
10299
more_args = "max_model_len=2048,max_num_seqs=128,kv_cache_dtype=fp8"
103100

104101
# Add TP test (if provided)

0 commit comments

Comments
 (0)