Modify autoencoderkl as per #1924.

Signed-off-by: Virginia Fernandez <[email protected]>

Author: Virginia

generation/3d_autoencoderkl/3d_autoencoderkl_tutorial.ipynb

@@ -178,6 +178,8 @@
    "metadata": {},
    "outputs": [],
    "source": [
+    "channel = 0  # 0 = Flair\n",
+    "assert channel in [0, 1, 2, 3], \"Choose a valid channel\"\n",
     "all_transforms = [\n",
     "    transforms.LoadImaged(keys=[\"image\"]),\n",
     "    transforms.EnsureChannelFirstd(keys=[\"image\"]),\n",
@@ -207,8 +209,6 @@
     }
    ],
    "source": [
-    "channel = 0  # 0 = Flair\n",
-    "assert channel in [0, 1, 2, 3], \"Choose a valid channel\"\n",
     "# Training\n",
     "train_ds = DecathlonDataset(\n",
     "    root_dir=root_dir,\n",
@@ -785,7 +785,7 @@
    "name": "python",
    "nbconvert_exporter": "python",
    "pygments_lexer": "ipython3",
-   "version": "3.12.0"
+   "version": "3.10.15"
   }
  },
  "nbformat": 4,