diff --git a/tests/inference/test_execution.py b/tests/inference/test_execution.py index e7b29302e..de66df9a1 100644 --- a/tests/inference/test_execution.py +++ b/tests/inference/test_execution.py @@ -14,6 +14,10 @@ import urllib.request import urllib.parse import urllib.error from comfy_execution.graph_utils import GraphBuilder, Node +import os +import sys + +PYTHON_EXECUTABLE_PATH = os.path.realpath(sys.executable) def run_warmup(client, prefix="warmup"): """Run a simple workflow to warm up the server.""" @@ -145,7 +149,7 @@ class TestExecution: def _server(self, args_pytest, request): # Start server pargs = [ - 'python','main.py', + PYTHON_EXECUTABLE_PATH,'main.py', '--output-directory', args_pytest["output_dir"], '--listen', args_pytest["listen"], '--port', str(args_pytest["port"]), diff --git a/tests/inference/test_inference.py b/tests/inference/test_inference.py index 7e4a206c4..bc7336f04 100644 --- a/tests/inference/test_inference.py +++ b/tests/inference/test_inference.py @@ -14,6 +14,8 @@ import websocket #NOTE: websocket-client (https://github.com/websocket-client/we import uuid import urllib.request import urllib.parse +import sys + from comfy.samplers import KSampler @@ -21,6 +23,7 @@ from comfy.samplers import KSampler """ These tests generate and save images through a range of parameters """ +PYTHON_EXECUTABLE_PATH = os.path.realpath(sys.executable) class ComfyGraph: def __init__(self, @@ -152,7 +155,7 @@ class TestInference: def _server(self, args_pytest): # Start server p = subprocess.Popen([ - 'python','main.py', + PYTHON_EXECUTABLE_PATH,'main.py', '--output-directory', args_pytest["output_dir"], '--listen', args_pytest["listen"], '--port', str(args_pytest["port"]),