apply sonar lint
This commit is contained in:
parent
17e25a3bc3
commit
192bd08156
|
@ -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,
|
||||
|
|
|
@ -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]:
|
||||
|
|
|
@ -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))
|
||||
|
||||
|
||||
|
|
|
@ -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)
|
||||
|
||||
|
|
|
@ -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,
|
||||
|
|
|
@ -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
|
||||
|
|
Loading…
Reference in New Issue