From 7adec4cf9e31a7e1a049dae6e0973604ff03aeae Mon Sep 17 00:00:00 2001 From: DarkLight1337 Date: Wed, 19 Jun 2024 12:42:03 +0000 Subject: [PATCH] Remove conftest change (it is in #5681 now) --- tests/conftest.py | 4 +--- 1 file changed, 1 insertion(+), 3 deletions(-) diff --git a/tests/conftest.py b/tests/conftest.py index 5bbfd87f0bb3b..f37c9883f7cf1 100644 --- a/tests/conftest.py +++ b/tests/conftest.py @@ -365,7 +365,7 @@ def __exit__(self, exc_type, exc_value, traceback): cleanup() -@pytest.fixture(scope="session") +@pytest.fixture def hf_runner(): return HfRunner @@ -385,7 +385,6 @@ def __init__( block_size: int = 16, enable_chunked_prefill: bool = False, swap_space: int = 4, - enforce_eager: bool = False, **kwargs, ) -> None: self.model = LLM( @@ -394,7 +393,6 @@ def __init__( trust_remote_code=True, dtype=dtype, swap_space=swap_space, - enforce_eager=enforce_eager, disable_log_stats=disable_log_stats, tensor_parallel_size=tensor_parallel_size, max_model_len=max_model_len,