From 9dc11b993380f67dd7b0a836c148ebe78eed1c7f Mon Sep 17 00:00:00 2001
From: doctorpangloss <@hiddenswitch.com>
Date: Fri, 14 Jun 2024 15:00:53 -0700
Subject: [PATCH] Fix inference tests

---
 tests/inference/test_inference.py | 7 +++----
 1 file changed, 3 insertions(+), 4 deletions(-)

diff --git a/tests/inference/test_inference.py b/tests/inference/test_inference.py
index 20dd3c235..7828f0009 100644
--- a/tests/inference/test_inference.py
+++ b/tests/inference/test_inference.py
@@ -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)