wangbo/ComfyUI
1
0
Fork 0
mirror of https://github.com/comfyanonymous/ComfyUI.git synced 2026-01-24 13:20:19 +08:00
Code Issues Actions 12 Packages Projects Releases Wiki Activity

Fix inference test setup bug

Browse Source
This commit is contained in:
doctorpangloss 2024-06-14 15:09:53 -07:00
parent 9dc11b9933
commit 0ace50c068
1 changed files with 2 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

3
tests/inference/test_inference.py
View File

@ -177,9 +177,10 @@ class TestInference:
# The "graph" is the default graph # The "graph" is the default graph
@fixture(scope="function", params=comfy_graph_list, ids=comfy_graph_ids, autouse=False) @fixture(scope="function", params=comfy_graph_list, ids=comfy_graph_ids, autouse=False)
def _client_graph(self, request, comfy_background_server) -> (ComfyClient, ComfyGraph): def _client_graph(self, request, comfy_background_server) -> (ComfyClient, ComfyGraph):
configuration, _ = comfy_background_server
comfy_graph = request.param comfy_graph = request.param
# Start client # Start client
comfy_client = self.start_client(comfy_background_server.listen, comfy_background_server.port) comfy_client = self.start_client(configuration.listen, configuration.port)
# Warm up pipeline # Warm up pipeline
comfy_client.get_images(graph=comfy_graph.graph, save=False) comfy_client.get_images(graph=comfy_graph.graph, save=False)