black style change

Author: arda-argmax

tests/torch2coreml/test_mmdit.py

@@ -61,7 +61,9 @@ def setUpClass(cls):
             .eval()
         )
         logger.info("Initialized.")
-        TEST_SD3_CKPT_PATH = TEST_SD3_CKPT_PATH or hf_hub_download(TEST_SD3_HF_REPO, "sd3_medium.safetensors")
+        TEST_SD3_CKPT_PATH = TEST_SD3_CKPT_PATH or hf_hub_download(
+            TEST_SD3_HF_REPO, "sd3_medium.safetensors"
+        )
         if TEST_SD3_CKPT_PATH is not None:
 
             logger.info(f"Loading SD3 model checkpoint from {TEST_SD3_CKPT_PATH}")
@@ -134,7 +136,9 @@ def get_test_inputs(cfg: mmdit.MMDiTConfig) -> Dict[str, torch.Tensor]:
     parser.add_argument("--latent-size", default=TEST_LATENT_SIZE, type=int)
     args = parser.parse_args()
 
-    TEST_SD3_CKPT_PATH = args.sd3_ckpt_path if os.path.exists(args.sd3_ckpt_path) else None
+    TEST_SD3_CKPT_PATH = (
+        args.sd3_ckpt_path if os.path.exists(args.sd3_ckpt_path) else None
+    )
     TEST_SD3_HF_REPO = args.sd3_ckpt_path
     TEST_LATENT_SIZE = args.latent_size
     TEST_CKPT_FILE_NAME = args.ckpt_file_name

tests/torch2coreml/test_vae.py

@@ -57,7 +57,9 @@ def setUpClass(cls):
         )
         logger.info("Initialized.")
 
-        TEST_SD3_CKPT_PATH = TEST_SD3_CKPT_PATH or hf_hub_download(TEST_SD3_HF_REPO, "sd3_medium.safetensors")
+        TEST_SD3_CKPT_PATH = TEST_SD3_CKPT_PATH or hf_hub_download(
+            TEST_SD3_HF_REPO, "sd3_medium.safetensors"
+        )
         if TEST_SD3_CKPT_PATH is not None:
             logger.info(f"Loading SD3 model checkpoint from {TEST_SD3_CKPT_PATH}")
             _load_vae_decoder_weights(cls.test_torch_model, TEST_SD3_CKPT_PATH)
@@ -104,7 +106,9 @@ def get_test_inputs(config: vae.VAEDecoderConfig) -> Dict[str, torch.Tensor]:
     parser.add_argument("--latent-size", default=TEST_LATENT_SIZE, type=int)
     args = parser.parse_args()
 
-    TEST_SD3_CKPT_PATH = args.sd3_ckpt_path if os.path.exists(args.sd3_ckpt_path) else None
+    TEST_SD3_CKPT_PATH = (
+        args.sd3_ckpt_path if os.path.exists(args.sd3_ckpt_path) else None
+    )
     TEST_SD3_HF_REPO = args.sd3_ckpt_path
     TEST_LATENT_SIZE = args.latent_size