wangbo/ComfyUI
1
0
Fork 0
mirror of https://github.com/comfyanonymous/ComfyUI.git synced 2026-01-11 23:00:51 +08:00
Code Issues Actions 10 Packages Projects Releases Wiki Activity

Merge branch 'hiddenswitch:master' into master

Browse Source
This commit is contained in:
Max Tretikov 2024-06-14 17:03:48 -06:00 committed by GitHub
commit 697d7603cb
No known key found for this signature in database
GPG Key ID: B5690EEEBB952194
1 changed files with 4 additions and 4 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

8
tests/inference/test_inference.py
View File

@ -175,12 +175,12 @@ class TestInference:
#
# Returns a "_client_graph", which is client-graph pair corresponding to an initialized server
# The "graph" is the default graph
@fixture(scope="class", params=comfy_graph_list, ids=comfy_graph_ids, autouse=False)
def _client_graph(self, request, args_pytest, comfy_background_server) -> (ComfyClient, ComfyGraph):
@fixture(scope="function", params=comfy_graph_list, ids=comfy_graph_ids, autouse=False)
def _client_graph(self, request, comfy_background_server) -> (ComfyClient, ComfyGraph):
configuration, _ = comfy_background_server
comfy_graph = request.param
# Start client
comfy_client = self.start_client(args_pytest["listen"], args_pytest["port"])
comfy_client = self.start_client(configuration.listen, configuration.port)
# Warm up pipeline
comfy_client.get_images(graph=comfy_graph.graph, save=False)