Skip to content

Commit 06e1e08

Browse files
[XPU] Fix test_logprobs_e2e import error: pin lm-eval[api]>=0.4.12 (#44469)
Signed-off-by: Chaojun Zhang <chaojun.zhang@intel.com>
1 parent 1a59078 commit 06e1e08

3 files changed

Lines changed: 6 additions & 9 deletions

File tree

.buildkite/intel_jobs/misc_intel.yaml

Lines changed: 4 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -57,14 +57,16 @@ steps:
5757
commands:
5858
- >-
5959
bash .buildkite/scripts/hardware_ci/run-intel-test.sh
60-
'export VLLM_WORKER_MULTIPROC_METHOD=spawn &&
60+
'pip install lm_eval[api]>=0.4.12 &&
61+
export VLLM_WORKER_MULTIPROC_METHOD=spawn &&
6162
cd tests &&
6263
pytest -v -s v1/logits_processors --ignore=v1/logits_processors/test_custom_online.py --ignore=v1/logits_processors/test_custom_offline.py &&
6364
pytest -v -s v1/test_oracle.py &&
6465
pytest -v -s v1/test_request.py &&
6566
pytest -v -s v1/test_outputs.py &&
6667
pytest -v -s v1/sample/test_topk_topp_sampler.py &&
67-
pytest -v -s v1/sample/test_logprobs.py'
68+
pytest -v -s v1/sample/test_logprobs.py &&
69+
pytest -v -s v1/sample/test_logprobs_e2e.py'
6870
6971
- label: XPU CPU Offload
7072
timeout_in_minutes: 60

requirements/test/xpu.in

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -13,7 +13,7 @@ pytest-shard
1313
absl-py
1414
accelerate
1515
arctic-inference
16-
lm_eval[api]
16+
lm_eval[api]>=0.4.12
1717
modelscope
1818

1919
# --- Audio Processing ---

requirements/test/xpu.txt

Lines changed: 1 addition & 6 deletions
Original file line numberDiff line numberDiff line change
@@ -33,7 +33,6 @@ arctic-inference==0.1.1
3333
attrs==26.1.0
3434
# via
3535
# aiohttp
36-
# jsonlines
3736
# jsonschema
3837
# referencing
3938
audioread==3.0.1
@@ -225,8 +224,6 @@ joblib==1.5.3
225224
# librosa
226225
# nltk
227226
# scikit-learn
228-
jsonlines==4.0.0
229-
# via lm-eval
230227
jsonschema==4.26.0
231228
# via
232229
# -c requirements/common.txt
@@ -247,7 +244,7 @@ librosa==0.10.2.post1
247244
# via -r requirements/test/xpu.in
248245
llvmlite==0.47.0
249246
# via numba
250-
lm-eval==0.4.11
247+
lm-eval==0.4.12
251248
# via -r requirements/test/xpu.in
252249
lxml==6.0.2
253250
# via
@@ -734,5 +731,3 @@ xxhash==3.6.0
734731
# evaluate
735732
yarl==1.23.0
736733
# via aiohttp
737-
zstandard==0.25.0
738-
# via lm-eval

0 commit comments

Comments
 (0)