From 0ace50c0680db46fdc79689735f01beaf93a45b5 Mon Sep 17 00:00:00 2001
From: doctorpangloss <@hiddenswitch.com>
Date: Fri, 14 Jun 2024 15:09:53 -0700
Subject: [PATCH] Fix inference test setup bug

---
 tests/inference/test_inference.py | 3 ++-
 1 file changed, 2 insertions(+), 1 deletion(-)

diff --git a/tests/inference/test_inference.py b/tests/inference/test_inference.py
index 7828f0009..05107bd2b 100644
--- a/tests/inference/test_inference.py
+++ b/tests/inference/test_inference.py
@@ -177,9 +177,10 @@ class TestInference:
     # The "graph" is the default graph
     @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(comfy_background_server.listen, comfy_background_server.port)
+        comfy_client = self.start_client(configuration.listen, configuration.port)
 
         # Warm up pipeline
         comfy_client.get_images(graph=comfy_graph.graph, save=False)