[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/maisi_diff_unet_training_tutorial.ipynb

@@ -51,8 +51,8 @@
    "metadata": {},
    "outputs": [],
    "source": [
-    "maisi_version = 'maisi-ddpm'\n",
-    "assert maisi_version in ['maisi-ddpm', 'maisi-rflow']"
+    "maisi_version = \"maisi-ddpm\"\n",
+    "assert maisi_version in [\"maisi-ddpm\", \"maisi-rflow\"]"
    ]
   },
   {
@@ -243,10 +243,10 @@
    "source": [
     "env_config_path = \"./configs/environment_maisi_diff_model.json\"\n",
     "model_config_path = \"./configs/config_maisi_diff_model.json\"\n",
-    "if maisi_version == 'maisi-ddpm':\n",
+    "if maisi_version == \"maisi-ddpm\":\n",
     "    model_def_path = \"./configs/config_maisi-ddpm.json\"\n",
     "    include_body_region = True\n",
-    "elif maisi_version == 'maisi-rflow':\n",
+    "elif maisi_version == \"maisi-rflow\":\n",
     "    model_def_path = \"./configs/config_maisi-rflow.json\"\n",
     "    include_body_region = False\n",
     "else:\n",
@@ -466,10 +466,7 @@
     "        spacing = [float(_item) for _item in spacing]\n",
     "\n",
     "        # Create the dictionary with the specified keys and values\n",
-    "        data = {\n",
-    "                \"dim\": dimensions,\n",
-    "                \"spacing\": spacing\n",
-    "        }\n",
+    "        data = {\"dim\": dimensions, \"spacing\": spacing}\n",
     "        if include_body_region:\n",
     "            # The region can be selected from one of four regions from top to bottom.\n",
     "            # [1,0,0,0] is the head and neck, [0,1,0,0] is the chest region, [0,0,1,0]\n",

generation/maisi/scripts/diff_model_infer.py

@@ -179,7 +179,7 @@ def run_inference(
                 image, _ = noise_scheduler.step(model_output, t, image, next_t)  # type: ignore
 
         inferer = SlidingWindowInferer(
-            roi_size=[80,80,80],
+            roi_size=[80, 80, 80],
             sw_batch_size=1,
             progress=True,
             mode="gaussian",