Skip to content

Commit 09f114d

Browse files
committed
Revert "[None][fix] restore pytest collection compatibility"
This reverts commit 5eb3254. Signed-off-by: Fanrong Li <lfr-0531@users.noreply.github.com>
1 parent 5eb3254 commit 09f114d

2 files changed

Lines changed: 3 additions & 3 deletions

File tree

tests/unittest/conftest.py

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -337,12 +337,12 @@ def torch_empty_cache() -> None:
337337
torch.cuda.empty_cache()
338338

339339

340-
@pytest.fixture(scope="module")
340+
@pytest.fixture(scope="module", params=[2, 4, 8])
341341
def mpi_pool_executor(request):
342342
"""
343343
Start an MPIPoolExecutor with `request.param` workers.
344344
"""
345-
num_workers = getattr(request, "param", 2)
345+
num_workers = request.param
346346
with MPIPoolExecutor(num_workers) as executor:
347347
# make the number of workers visible to tests
348348
setattr(executor, "num_workers", num_workers)

tests/unittest/llmapi/test_llm_pytorch.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -67,7 +67,7 @@
6767

6868

6969
@force_ampere
70-
@pytest.mark.parametrize("enable_chunked_prefill", [False, True])
70+
@pytest.mark.parametrize("enable_chunked_prefill,", [False, True])
7171
@pytest.mark.part2
7272
def test_tinyllama_logits_processor(enable_chunked_prefill):
7373
tinyllama_logits_processor_test_harness(

0 commit comments

Comments
 (0)