Skip to content

Commit 79c1b83

Browse files
committed
Revert "Set VLLM_PORT env variable"
1 parent ede93a9 commit 79c1b83

File tree

1 file changed

+0
-6
lines changed

1 file changed

+0
-6
lines changed

tests/utils.py

-6
Original file line numberDiff line numberDiff line change
@@ -179,7 +179,6 @@ def compare_two_settings(model: str,
179179
env1: The first set of environment variables to pass to the API server.
180180
env2: The second set of environment variables to pass to the API server.
181181
"""
182-
os.environ["VLLM_PORT"] = "8001"
183182

184183
trust_remote_code = "--trust-remote-code"
185184
if trust_remote_code in arg1 or trust_remote_code in arg2:
@@ -299,8 +298,6 @@ def compare_two_settings(model: str,
299298
"texts": texts,
300299
})
301300

302-
os.environ.pop("VLLM_PORT")
303-
304301
n = len(results) // 2
305302
arg1_results = results[:n]
306303
arg2_results = results[n:]
@@ -494,7 +491,6 @@ async def completions_with_server_args(
494491
Returns:
495492
OpenAI Completion instance
496493
'''
497-
os.environ["VLLM_PORT"] = "8001"
498494

499495
outputs = None
500496
with RemoteOpenAIServer(model_name,
@@ -507,8 +503,6 @@ async def completions_with_server_args(
507503
stream=False,
508504
max_tokens=5,
509505
logprobs=num_logprobs)
510-
os.environ.pop("VLLM_PORT")
511-
512506
assert outputs is not None
513507

514508
return outputs

0 commit comments

Comments
 (0)