diff --git a/_local_envs_for_test.sh b/_local_envs_for_test.sh index 098d87d8..829fd8cf 100644 --- a/_local_envs_for_test.sh +++ b/_local_envs_for_test.sh @@ -14,7 +14,6 @@ export VLLM_SPYRE_DYNAMO_BACKEND=eager # This makes debugging easy on local setup export VLLM_ENABLE_V1_MULTIPROCESSING=0 -export VLLM_SPYRE_TEST_BACKEND_LIST=eager # We have to use `HF_HUB_OFFLINE=1` otherwise vllm might try to download a # different version of the model using HF API which might not work locally export HF_HUB_OFFLINE=1 diff --git a/tests/spyre_util.py b/tests/spyre_util.py index 8e95212a..203edbed 100644 --- a/tests/spyre_util.py +++ b/tests/spyre_util.py @@ -197,14 +197,12 @@ def get_spyre_model_dir_path() -> Path: return Path(model_dir_path) -# get model backends from env or default to all and add pytest markers +# add pytest markers to supported different backends def get_spyre_backend_list(): - user_backend_list = os.environ.get("VLLM_SPYRE_TEST_BACKEND_LIST", - "eager,inductor,sendnn") + backend_list = ["eager", "inductor", "sendnn"] backends = [] - for backend in user_backend_list.split(","): - backend = backend.strip() + for backend in backend_list: marks = [] if backend == "eager": marks = [pytest.mark.cpu] diff --git a/tests/utils/test_spyre_backend_list.py b/tests/utils/test_spyre_backend_list.py deleted file mode 100644 index 7b818c0a..00000000 --- a/tests/utils/test_spyre_backend_list.py +++ /dev/null @@ -1,21 +0,0 @@ -import pytest -from spyre_util import get_spyre_backend_list - - -@pytest.mark.utils -@pytest.mark.cpu -def test_get_spyre_backend_list(monkeypatch): - ''' - Ensure we return the backend list correctly - ''' - with monkeypatch.context() as m: - m.setenv("VLLM_SPYRE_TEST_BACKEND_LIST", "eager,inductor,sendnn") - backend_list = get_spyre_backend_list() - assert backend_list[0].values[0] == "eager" - assert backend_list[1].values[0] == "inductor" - assert backend_list[2].values[0] == "sendnn" - - with monkeypatch.context() as m: - m.setenv("VLLM_SPYRE_TEST_BACKEND_LIST", "sendnn") - backend_list = get_spyre_backend_list() - assert backend_list[0].values[0] == "sendnn"