`Autofix tests

Virginia Fernandez · Virginia Fernandez · commit 59cbe0dd284c · 2025-09-08T14:59:39.000+01:00
diff --git a/tests/inferers/test_diffusion_inferer.py b/tests/inferers/test_diffusion_inferer.py
@@ -106,7 +106,7 @@ def test_sample_cfg(self, model_params, input_shape):
             save_intermediates=True,
             intermediate_steps=1,
             cfg=5,
-            cfg_fill_value=-1
+            cfg_fill_value=-1,
         )
         self.assertEqual(sample.shape, noise.shape)
 
diff --git a/tests/inferers/test_latent_diffusion_inferer.py b/tests/inferers/test_latent_diffusion_inferer.py
@@ -456,7 +456,7 @@ def test_sample_shape_with_cfg(
                     scheduler=scheduler,
                     seg=input_seg,
                     cfg=5,
-                    cfg_fill_value=-1
+                    cfg_fill_value=-1,
                 )
             else:
                 sample = inferer.sample(

Original file line number	Diff line number	Diff line change
`@@ -106,7 +106,7 @@ def test_sample_cfg(self, model_params, input_shape):`
`106`	`106`	`save_intermediates=True,`
`107`	`107`	`intermediate_steps=1,`
`108`	`108`	`cfg=5,`
`109`		`- cfg_fill_value=-1`
	`109`	`+ cfg_fill_value=-1,`
`110`	`110`	`)`
`111`	`111`	`self.assertEqual(sample.shape, noise.shape)`
`112`	`112`
Original file line number	Diff line number	Diff line change
`@@ -456,7 +456,7 @@ def test_sample_shape_with_cfg(`
`456`	`456`	`scheduler=scheduler,`
`457`	`457`	`seg=input_seg,`
`458`	`458`	`cfg=5,`
`459`		`- cfg_fill_value=-1`
	`459`	`+ cfg_fill_value=-1,`
`460`	`460`	`)`
`461`	`461`	`else:`
`462`	`462`	`sample = inferer.sample(`