Skip to content

Commit

Permalink
Remove superfluous working_dir=VLLM_PATH in ray.init
Browse files Browse the repository at this point in the history
  • Loading branch information
mawong-amd committed Jun 29, 2024
1 parent 907f000 commit d1c07b5
Showing 1 changed file with 2 additions and 2 deletions.
4 changes: 2 additions & 2 deletions tests/entrypoints/test_openai_chat.py
Original file line number Diff line number Diff line change
Expand Up @@ -14,7 +14,7 @@
from huggingface_hub import snapshot_download
from openai import BadRequestError

from ..utils import VLLM_PATH, RemoteOpenAIServer
from ..utils import RemoteOpenAIServer

# any model with a chat template should work here
MODEL_NAME = "HuggingFaceH4/zephyr-7b-beta"
Expand Down Expand Up @@ -79,7 +79,7 @@ def zephyr_lora_files():

@pytest.fixture(scope="module")
def ray_ctx():
ray.init(runtime_env={"working_dir": VLLM_PATH})
ray.init()
yield
ray.shutdown()

Expand Down

0 comments on commit d1c07b5

Please sign in to comment.