diff --git a/tests/test_entrypoint/test_entrypoint.py b/tests/test_entrypoint/test_entrypoint.py index 95703bd..0924eca 100644 --- a/tests/test_entrypoint/test_entrypoint.py +++ b/tests/test_entrypoint/test_entrypoint.py @@ -2,6 +2,7 @@ import os import shutil import subprocess +import sys import time import unittest @@ -61,7 +62,7 @@ def test_entrypoint(self): input_stream = json.dumps(input_data) main_file = self._file_name("main.py", self.TWO_DIRS_UP) - args = ["python", main_file] + args = [sys.executable, main_file] result = subprocess.run( args, env=os.environ,