Add ruff rule for empty line with trailing whitespace.

tests/compare/test_quality.py

@@ -22,7 +22,7 @@ def ssim_score(img0: np.ndarray, img1: np.ndarray) -> Tuple[float, np.ndarray]:
     # rescale the difference image to 0-255 range
     diff = (diff * 255).astype("uint8")
     return score, diff
-    
+
 # Metrics must return a tuple of (score, diff_image)
 METRICS = {"ssim": ssim_score}
 METRICS_PASS_THRESHOLD = {"ssim": 0.95}
@@ -64,13 +64,13 @@ class TestCompareImageMetrics:
                     image_list = [[Image.open(file) for file in files] for files in image_file_list]
                     grid = self.image_grid(image_list)
                     grid.save(os.path.join(grid_dir, f"{metric_dir}_{score:.3f}_{file}"))
-    
+
     # Tests run for each baseline file name
     @fixture()
     def fname(self, baseline_fname):
         yield baseline_fname
         del baseline_fname
-    
+
     def test_directories_not_empty(self, args_pytest):
         baseline_dir = args_pytest['baseline_dir']
         test_dir = args_pytest['test_dir']
@@ -98,7 +98,7 @@ class TestCompareImageMetrics:
         test_dir = args_pytest['test_dir']
         metrics_output_file = args_pytest['metrics_file']
         img_output_dir = args_pytest['img_output_dir']
-        
+
         baseline_file_path = os.path.join(baseline_dir, fname)
 
         # Find file match
@@ -108,7 +108,7 @@ class TestCompareImageMetrics:
         # Run metrics
         sample_baseline = self.read_img(baseline_file_path)
         sample_secondary = self.read_img(test_file)
-        
+
         score, metric_img = METRICS[metric](sample_baseline, sample_secondary)
         metric_status = score > METRICS_PASS_THRESHOLD[metric]
 
@@ -140,7 +140,7 @@ class TestCompareImageMetrics:
 
         w, h = img_list[0][0].size
         grid = Image.new('RGB', size=(cols*w, rows*h))
-        
+
         for i, row in enumerate(img_list):
             for j, img in enumerate(row):
                 grid.paste(img, box=(j*w, i*h))
@@ -170,7 +170,7 @@ class TestCompareImageMetrics:
         img = Image.open(fname)
         img.load()
         return img.info['prompt']
-    
+
     def find_file_match(self, baseline_file: str, file_paths: List[str]):
         # Find a file in file_paths with matching metadata to baseline_file
         baseline_prompt = self.read_file_prompt(baseline_file)
@@ -192,4 +192,4 @@ class TestCompareImageMetrics:
         for f in file_paths:
             test_file_prompt = self.read_file_prompt(f)
             if baseline_prompt == test_file_prompt:
-                return f
+                return f

tests/conftest.py

@@ -21,7 +21,7 @@ def args_pytest(pytestconfig):
 
 def pytest_collection_modifyitems(items):
     # Modifies items so tests run in the correct order
-    
+
     LAST_TESTS = ['test_quality']
 
     # Move the last items to the end

tests/inference/test_execution.py

@@ -368,7 +368,7 @@ class TestExecution:
         g.node("SaveImage", images=mix1.out(0))
         g.node("SaveImage", images=mix2.out(0))
         g.remove_node("removeme")
-        
+
         client.run(g)
 
         # Add back in the missing node to make sure the error doesn't break the server
@@ -512,7 +512,7 @@ class TestExecution:
         int_list = g.node("TestMakeListNode", value1=int1.out(0), value2=int2.out(0), value3=int3.out(0))
         compare = g.node("TestIntConditions", a=int_list.out(0), b=2, operation="==")
         blocker = g.node("TestExecutionBlocker", input=image_list.out(0), block=compare.out(0), verbose=False)
-        
+
         list_output = g.node("TestMakeListNode", value1=blocker.out(0))
         output = g.node("PreviewImage", images=list_output.out(0))
 

tests/inference/test_inference.py

@@ -43,12 +43,12 @@ class ComfyGraph:
         # sets the sampler name for the sampler nodes (eg. base and refiner)
         for node in self.sampler_nodes:
             self.graph[node]['inputs']['sampler_name'] = sampler_name
-    
+
     def set_scheduler(self, scheduler:str):
         # sets the sampler name for the sampler nodes (eg. base and refiner)
         for node in self.sampler_nodes:
             self.graph[node]['inputs']['scheduler'] = scheduler
-    
+
     def set_filename_prefix(self, prefix:str):
         # sets the filename prefix for the save nodes
         for node in self.graph:
@@ -185,7 +185,7 @@ class TestInference:
     @fixture(scope="class", params=comfy_graph_list, ids=comfy_graph_ids, autouse=True)
     def _client_graph(self, request, args_pytest, _server) -> (ComfyClient, ComfyGraph):
         comfy_graph = request.param
-        
+
         # Start client
         comfy_client = self.start_client(args_pytest["listen"], args_pytest["port"])
 
@@ -201,7 +201,7 @@ class TestInference:
     def client(self, _client_graph):
         client = _client_graph[0]
         yield client
-    
+
     @fixture
     def comfy_graph(self, _client_graph):
         # avoid mutating the graph

tests/inference/testing_nodes/testing-pack/specific_tests.py

@@ -87,7 +87,7 @@ class TestCustomIsChanged:
 
     def custom_is_changed(self, image, should_change=False):
         return (image,)
-    
+
     @classmethod
     def IS_CHANGED(cls, should_change=False, *args, **kwargs):
         if should_change:
@@ -112,7 +112,7 @@ class TestIsChangedWithConstants:
 
     def custom_is_changed(self, image, value):
         return (image * value,)
-    
+
     @classmethod
     def IS_CHANGED(cls, image, value):
         if image is None:

tests/inference/testing_nodes/testing-pack/util.py

@@ -145,7 +145,7 @@ class TestAccumulationGetLengthNode:
 
     def accumlength(self, accumulation):
         return (len(accumulation['accum']),)
-        
+
 @VariantSupport()
 class TestAccumulationGetItemNode:
     def __init__(self):
@@ -168,7 +168,7 @@ class TestAccumulationGetItemNode:
 
     def get_item(self, accumulation, index):
         return (accumulation['accum'][index],)
-        
+
 @VariantSupport()
 class TestAccumulationSetItemNode:
     def __init__(self):