From 69cdc634e490aeb6ea0c7c513e09005076640082 Mon Sep 17 00:00:00 2001 From: Alexander Eichhorn Date: Mon, 20 Apr 2026 17:50:24 +0200 Subject: [PATCH] Chore Ruff --- invokeai/app/invocations/image.py | 4 +--- tests/app/invocations/test_save_image_to_file.py | 13 +++++++------ 2 files changed, 8 insertions(+), 9 deletions(-) diff --git a/invokeai/app/invocations/image.py b/invokeai/app/invocations/image.py index ef00077d9c..1cb7f6d973 100644 --- a/invokeai/app/invocations/image.py +++ b/invokeai/app/invocations/image.py @@ -1065,9 +1065,7 @@ class SaveImageToFileInvocation(BaseInvocation, WithMetadata, WithBoard): candidate = (outputs_path / raw).resolve() outputs_resolved = outputs_path.resolve() if outputs_resolved != candidate and outputs_resolved not in candidate.parents: - raise ValueError( - f"output_directory must stay within the outputs folder: {self.output_directory!r}" - ) + raise ValueError(f"output_directory must stay within the outputs folder: {self.output_directory!r}") target_dir = candidate target_dir.mkdir(parents=True, exist_ok=True) diff --git a/tests/app/invocations/test_save_image_to_file.py b/tests/app/invocations/test_save_image_to_file.py index 1c8684e2d1..095f67be28 100644 --- a/tests/app/invocations/test_save_image_to_file.py +++ b/tests/app/invocations/test_save_image_to_file.py @@ -5,6 +5,7 @@ from unittest.mock import MagicMock import pytest from PIL import Image +from pydantic import ValidationError from invokeai.app.invocations.image import SaveImageToFileInvocation @@ -22,10 +23,10 @@ def _make_context(tmp_path: Path, pil_image: Image.Image, gallery_uuid: str = "a def _build_node(**overrides) -> SaveImageToFileInvocation: - defaults = dict( - id="test", - image={"image_name": "input.png"}, - ) + defaults = { + "id": "test", + "image": {"image_name": "input.png"}, + } defaults.update(overrides) return SaveImageToFileInvocation(**defaults) @@ -179,9 +180,9 @@ class TestSaveImageToFileInvocation: assert saved.format == "WEBP" def test_quality_bounds_enforced_by_pydantic(self): - with pytest.raises(Exception): + with pytest.raises(ValidationError): _build_node(quality=0) - with pytest.raises(Exception): + with pytest.raises(ValidationError): _build_node(quality=101) def test_output_is_pass_through_of_gallery_dto(self, tmp_path):