remedy missed ruff check/formats in 0.6.0 release

src/mflux/callbacks/callback_registry.py

@@ -1,4 +1,4 @@
-from mflux.callbacks.callback import BeforeLoopCallback, InLoopCallback, InterruptCallback, AfterLoopCallback
+from mflux.callbacks.callback import AfterLoopCallback, BeforeLoopCallback, InLoopCallback, InterruptCallback
 
 
 class CallbackRegistry:

src/mflux/callbacks/callbacks.py

@@ -51,12 +51,7 @@ class Callbacks:
         config: RuntimeConfig
     ):  # fmt: off
         for subscriber in CallbackRegistry.after_loop_callbacks():
-            subscriber.call_after_loop(
-                seed=seed,
-                prompt=prompt,
-                latents=latents,
-                config=config
-            )
+            subscriber.call_after_loop(seed=seed, prompt=prompt, latents=latents, config=config)
 
     @staticmethod
     def interruption(

src/mflux/callbacks/instances/memory_saver.py

@@ -4,11 +4,7 @@ import mlx.core as mx
 import PIL.Image
 from tqdm import tqdm
 
-from mflux.callbacks.callback import (
-    AfterLoopCallback,
-    BeforeLoopCallback,
-    InLoopCallback
-)
+from mflux.callbacks.callback import AfterLoopCallback, BeforeLoopCallback, InLoopCallback
 from mflux.config.runtime_config import RuntimeConfig
 
 
@@ -65,7 +61,7 @@ class MemorySaver(BeforeLoopCallback, InLoopCallback, AfterLoopCallback):
 
     def _delete_transformer(self) -> None:
         self.flux.transformer = None
-        if hasattr(self.flux, 'transformer_controlnet'):
+        if hasattr(self.flux, "transformer_controlnet"):
             self.flux.transformer_controlnet = None
         gc.collect()
         mx.metal.clear_cache()

src/mflux/generate.py

@@ -1,7 +1,7 @@
 from mflux import Config, Flux1, ModelConfig, StopImageGenerationException
 from mflux.callbacks.callback_registry import CallbackRegistry
-from mflux.callbacks.instances.stepwise_handler import StepwiseHandler
 from mflux.callbacks.instances.memory_saver import MemorySaver
+from mflux.callbacks.instances.stepwise_handler import StepwiseHandler
 from mflux.ui.cli.parsers import CommandLineParser
 
 

src/mflux/generate_controlnet.py

@@ -1,8 +1,8 @@
 from mflux import Config, Flux1Controlnet, ModelConfig, StopImageGenerationException
 from mflux.callbacks.callback_registry import CallbackRegistry
 from mflux.callbacks.instances.canny_saver import CannyImageSaver
-from mflux.callbacks.instances.stepwise_handler import StepwiseHandler
 from mflux.callbacks.instances.memory_saver import MemorySaver
+from mflux.callbacks.instances.stepwise_handler import StepwiseHandler
 from mflux.ui.cli.parsers import CommandLineParser
 
 
@@ -66,5 +66,6 @@ def main():
         if memory_saver:
             print(memory_saver.memory_stats())
 
+
 if __name__ == "__main__":
     main()

src/mflux/ui/cli/parsers.py

@@ -17,7 +17,7 @@ class ModelSpecAction(argparse.Action):
 
         if values.count("/") != 1:
             raise argparse.ArgumentError(
-                self, 'Value must be either "dev", "schnell", or "' f'in format "org/model". Got: {values}'
+                self, f'Value must be either "dev", "schnell", or "in format "org/model". Got: {values}'
             )
 
         # If we got here, values contains exactly one slash

tools/rename_images.py

@@ -51,10 +51,10 @@ def parse_metadata_from_image(image: Image, insecure=False) -> dict:
                     metadata = obj
                 else:
                     raise UnsupportedMetadata(
-                        "The metadata is not a dict output recognized by this tool. " f"Metadata: {value.decode()}"
+                        f"The metadata is not a dict output recognized by this tool. Metadata: {value.decode()}"
                     )
             except (ValueError, SyntaxError):
-                raise UnsupportedMetadata("The metadata is not parseable by this tool. " f"Metadata: {value.decode()}")
+                raise UnsupportedMetadata(f"The metadata is not parseable by this tool. Metadata: {value.decode()}")
 
     return metadata