[pre-commit.ci] auto fixes from pre-commit.com hooks

for more information, see https://pre-commit.ci

Author: pre-commit-ci[bot]

generation/maisi/scripts/download_model_data.py

@@ -87,7 +87,7 @@ def fetch_to_hf_path_cmd(
 
 def download_model_data(generate_version, root_dir, model_only=False):
     # TODO: remove the `files` after the files are uploaded to the NGC
-    if generate_version == 'maisi3d-ddpm' or generate_version == 'maisi3d-rflow':
+    if generate_version == "maisi3d-ddpm" or generate_version == "maisi3d-rflow":
         files = [
             {
                 "path": "models/autoencoder_v1.pt",
@@ -122,7 +122,7 @@ def download_model_data(generate_version, root_dir, model_only=False):
         raise ValueError(
             f"generate_version has to be chosen from ['maisi3d-ddpm', 'maisi3d-rflow'], yet got {generate_version}."
         )
-    if generate_version == 'maisi3d-ddpm':
+    if generate_version == "maisi3d-ddpm":
         files += [
             {
                 "path": "models/diff_unet_3d_ddpm-ct.pt",
@@ -143,7 +143,7 @@ def download_model_data(generate_version, root_dir, model_only=False):
                     "filename": "datasets/candidate_masks_flexible_size_and_spacing_3000.json",
                 },
             ]
-    elif generate_version == 'maisi3d-rflow':
+    elif generate_version == "maisi3d-rflow":
         files += [
             {
                 "path": "models/diff_unet_3d_rflow-ct.pt",
@@ -180,7 +180,7 @@ def download_model_data(generate_version, root_dir, model_only=False):
     parser.add_argument(
         "--version",
         type=str,
-        default='maisi3d-rflow',
+        default="maisi3d-rflow",
     )
     parser.add_argument(
         "--root_dir",