Chore Ruff

This commit is contained in:
Alexander Eichhorn
2026-04-20 17:50:24 +02:00
parent 209683d1de
commit 69cdc634e4
2 changed files with 8 additions and 9 deletions

View File

@@ -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)

View File

@@ -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):