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]>
apache · Nov 7, 2024 · 6742499 · 6742499
1 parent c7e4db7
commit 6742499
Showing 1 changed file with 2 additions and 1 deletion.
diff --git 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 logging
 import os
 import subprocess
+import sys
 import threading
 import time
 import uuid
@@ -118,7 +119,7 @@ def __init__(self, model_name: str, vllm_server_kwargs: Dict[str, str]):
   def start_server(self, retries=3):
     if not self._server_started:
       server_cmd = [
-          'python',
+          sys.executable,
           '-m',
           'vllm.entrypoints.openai.api_server',
           '--model',