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

Fix inference tests

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

7
tests/inference/test_inference.py
View File

@ -175,12 +175,11 @@ 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):
comfy_graph = request.param
# Start client
comfy_client = self.start_client(args_pytest["listen"], args_pytest["port"])
comfy_client = self.start_client(comfy_background_server.listen, comfy_background_server.port)
# Warm up pipeline
comfy_client.get_images(graph=comfy_graph.graph, save=False)