apply sonar lint

api/onnx_web/chain/pipeline.py

@@ -218,7 +218,7 @@ class ChainPipeline:
                     "image does not contain sources and is within tile size of %s, running stage",
                     tile,
                 )
-                for i in range(worker.retries):
+                for _i in range(worker.retries):
                     try:
                         stage_result = stage_pipe.run(
                             worker,

api/onnx_web/diffusers/load.py

@@ -95,7 +95,7 @@ def add_pipeline(name: str, pipeline: Any) -> bool:
         return False
     else:
         available_pipelines[name] = pipeline
-        return False
+        return True
 
 
 def get_available_pipelines() -> List[str]:

api/onnx_web/utils.py

@@ -41,7 +41,7 @@ def get_boolean(args: Any, key: str, default_value: bool) -> bool:
     return val.lower() in ("1", "t", "true", "y", "yes")
 
 
-def get_list(args: Any, key: str, default = "") -> List[str]:
+def get_list(args: Any, key: str, default="") -> List[str]:
     return split_list(args.get(key, default))
 
 

api/tests/convert/diffusion/test_lora.py

@@ -141,7 +141,7 @@ class FixXLNameTests(unittest.TestCase):
         nodes = {
             "lora_unet.input_block.lora_down.weight": {},
         }
-        fixed = fix_xl_names(nodes, [""])
+        fixed = fix_xl_names(nodes, [NodeProto(name="test")])
 
         self.assertEqual(fixed, nodes)
 

api/tests/test_diffusers/test_run.py

@@ -15,6 +15,9 @@ from onnx_web.server.context import ServerContext
 from onnx_web.worker.context import WorkerContext
 from tests.helpers import TEST_MODEL_DIFFUSION_SD15, test_device, test_needs_models
 
+TEST_PROMPT = "an astronaut eating a hamburger"
+TEST_SCHEDULER = "ddim"
+
 
 class TestTxt2ImgPipeline(unittest.TestCase):
     @test_needs_models([TEST_MODEL_DIFFUSION_SD15])
@@ -46,8 +49,8 @@ class TestTxt2ImgPipeline(unittest.TestCase):
             ImageParams(
                 TEST_MODEL_DIFFUSION_SD15,
                 "txt2img",
-                "ddim",
-                "an astronaut eating a hamburger",
+                TEST_SCHEDULER,
+                TEST_PROMPT,
                 3.0,
                 1,
                 1,
@@ -92,8 +95,8 @@ class TestTxt2ImgPipeline(unittest.TestCase):
             ImageParams(
                 TEST_MODEL_DIFFUSION_SD15,
                 "txt2img",
-                "ddim",
-                "an astronaut eating a hamburger",
+                TEST_SCHEDULER,
+                TEST_PROMPT,
                 3.0,
                 1,
                 1,
@@ -141,8 +144,8 @@ class TestTxt2ImgPipeline(unittest.TestCase):
             ImageParams(
                 TEST_MODEL_DIFFUSION_SD15,
                 "txt2img",
-                "ddim",
-                "an astronaut eating a hamburger",
+                TEST_SCHEDULER,
+                TEST_PROMPT,
                 3.0,
                 1,
                 1,
@@ -186,8 +189,8 @@ class TestTxt2ImgPipeline(unittest.TestCase):
             ImageParams(
                 TEST_MODEL_DIFFUSION_SD15,
                 "txt2img",
-                "ddim",
-                "an astronaut eating a hamburger",
+                TEST_SCHEDULER,
+                TEST_PROMPT,
                 3.0,
                 1,
                 1,
@@ -237,8 +240,8 @@ class TestImg2ImgPipeline(unittest.TestCase):
             ImageParams(
                 TEST_MODEL_DIFFUSION_SD15,
                 "txt2img",
-                "ddim",
-                "an astronaut eating a hamburger",
+                TEST_SCHEDULER,
+                TEST_PROMPT,
                 3.0,
                 1,
                 1,
@@ -284,8 +287,8 @@ class TestUpscalePipeline(unittest.TestCase):
             ImageParams(
                 "../models/upscaling-stable-diffusion-x4",
                 "txt2img",
-                "ddim",
-                "an astronaut eating a hamburger",
+                TEST_SCHEDULER,
+                TEST_PROMPT,
                 3.0,
                 1,
                 1,
@@ -331,8 +334,8 @@ class TestBlendPipeline(unittest.TestCase):
             ImageParams(
                 TEST_MODEL_DIFFUSION_SD15,
                 "txt2img",
-                "ddim",
-                "an astronaut eating a hamburger",
+                TEST_SCHEDULER,
+                TEST_PROMPT,
                 3.0,
                 1,
                 1,

api/tests/worker/test_worker.py

@@ -67,7 +67,6 @@ class WorkerMainTests(unittest.TestCase):
         )
 
         self.assertEqual(status, EXIT_INTERRUPT)
-        pass
 
     def test_pending_exception_retry(self):
         status = None
@@ -103,7 +102,6 @@ class WorkerMainTests(unittest.TestCase):
         )
 
         self.assertEqual(status, EXIT_ERROR)
-        pass
 
     def test_pending_exception_value(self):
         status = None