From 0e0856ffcbcbf75f9c41b3dd571b448e438926ff Mon Sep 17 00:00:00 2001 From: Austin Mroz Date: Thu, 27 Jun 2024 14:01:41 -0500 Subject: [PATCH] Explicitly define args for wrapped server start --- orchestration.py | 8 +++----- 1 file changed, 3 insertions(+), 5 deletions(-) diff --git a/orchestration.py b/orchestration.py index 57c3f21..fade372 100644 --- a/orchestration.py +++ b/orchestration.py @@ -8,16 +8,14 @@ ps = server.PromptServer.instance finished_startup = False original_server_start = ps.start -async def server_start(*args, **kwargs): - args = list(args) - original_on_start= args[3] +async def server_start(address, port, verbose=True, call_on_start=None): + original_on_start= call_on_start def on_start(scheme, address, port): if original_on_start is not None: original_on_start(scheme, address, port) global finished_startup finished_startup= True - args[3] = on_start - return await original_server_start(*args, **kwargs) + return await original_server_start(address, port, verbose, on_start) ps.start = server_start