formatting changes ruff formatting

Author: sfisher

ai_diffusion/text.py

@@ -1,7 +1,8 @@
 from __future__ import annotations
 import re
 from pathlib import Path
-from typing import Tuple, List, NamedTuple
+from typing import Tuple, List, NamedTuple, Optional
+
 
 from .api import LoraInput
 from .files import FileCollection, FileSource
@@ -230,6 +231,7 @@ def edit_attention(text: str, positive: bool) -> str:
         else f"{open_bracket}{attention_string}:{weight:.1f}{close_bracket}"
     )
 
+
 # in case a custom sampler is used then it may show steps and cfg in there
 def _parse_sampler_embeds(sampler: str) -> Tuple[str, Optional[int], Optional[float]]:
     """
@@ -266,6 +268,7 @@ def _parse_sampler_embeds(sampler: str) -> Tuple[str, Optional[int], Optional[fl
 
     return base, steps_in, cfg_in
 
+
 # creates the img text metadata for embedding in PNG files in style like Automatic1111
 def create_img_metadata(params: JobParams):
     meta = params.metadata

tests/test_text.py

@@ -5,7 +5,7 @@
     select_on_cursor_pos,
     create_img_metadata,
     strip_prompt_comments,
-    _parse_sampler_embeds
+    _parse_sampler_embeds,
 )
 from ai_diffusion.api import LoraInput
 from ai_diffusion.files import File, FileCollection
@@ -222,6 +222,7 @@ def test_create_img_metadata_missing_metadata_fields():
     assert "Size: 100x200" in result
     assert "Model: Unknown" in result
 
+
 def test_create_img_metadata_sampler_embedded():
     bounds = Bounds(0, 0, 832, 1216)
     metadata = {
@@ -247,6 +248,7 @@ def test_create_img_metadata_sampler_embedded():
     assert "Size: 832x1216" in result
     assert "Denoising strength" not in result
 
+
 def test_parse_sampler_embeds():
     assert _parse_sampler_embeds("") == ("", None, None)
     assert _parse_sampler_embeds("Euler") == ("Euler", None, None)
@@ -255,6 +257,7 @@ def test_parse_sampler_embeds():
     assert _parse_sampler_embeds("LCM (13)") == ("LCM", 13, None)
     assert _parse_sampler_embeds("LCM (13,1.1)") == ("LCM", 13, 1.1)
 
+
 class TestEditAttention:
     def test_empty_selection(self):
         assert edit_attention("", positive=True) == ""