diff --git a/adetailer/ultralytics.py b/adetailer/ultralytics.py index c63af40..1bd13ff 100644 --- a/adetailer/ultralytics.py +++ b/adetailer/ultralytics.py @@ -22,7 +22,7 @@ def ultralytics_predict( device: str = "", classes: str = "", ) -> PredictOutput[float]: - from ultralytics import YOLO # noqa: PLC0415 + from ultralytics import YOLO model = YOLO(model_path) apply_classes(model, model_path, classes) diff --git a/pyproject.toml b/pyproject.toml index a801b07..87fb281 100644 --- a/pyproject.toml +++ b/pyproject.toml @@ -74,7 +74,7 @@ select = [ "UP", "W", ] -ignore = ["B905", "E501", "PLR2004", "PLW0603"] +ignore = ["B905", "E501", "PLC0415", "PLR2004", "PLW0603"] unfixable = ["F401"] [tool.ruff.lint.isort]