This is an automated email from the ASF dual-hosted git repository.
tvalentyn pushed a commit to branch master
in repository https://gitbox.apache.org/repos/asf/beam.git
The following commit(s) were added to refs/heads/master by this push:
new 674249919c5 Use sys.executable to find python command. (#33033)
674249919c5 is described below
commit 674249919c51236041b635a3f864a5ba595d916e
Author: tvalentyn <[email protected]>
AuthorDate: Thu Nov 7 11:54:34 2024 -0800
Use sys.executable to find python command. (#33033)
* Use sys.executable to find python command.
* Make linter happy
---------
Co-authored-by: Danny McCormick <[email protected]>
---
sdks/python/apache_beam/ml/inference/vllm_inference.py | 3 ++-
1 file changed, 2 insertions(+), 1 deletion(-)
diff --git a/sdks/python/apache_beam/ml/inference/vllm_inference.py
b/sdks/python/apache_beam/ml/inference/vllm_inference.py
index b86d33ec16b..799083d16ce 100644
--- a/sdks/python/apache_beam/ml/inference/vllm_inference.py
+++ b/sdks/python/apache_beam/ml/inference/vllm_inference.py
@@ -21,6 +21,7 @@ import asyncio
import logging
import os
import subprocess
+import sys
import threading
import time
import uuid
@@ -118,7 +119,7 @@ class _VLLMModelServer():
def start_server(self, retries=3):
if not self._server_started:
server_cmd = [
- 'python',
+ sys.executable,
'-m',
'vllm.entrypoints.openai.api_server',
'--model',
