Incorporate suggestions made by Eric (fix of tempfile import and use,…

… typo). Signed-off-by: Virginia Fernandez <[email protected]>
Project-MONAI · Sep 11, 2024 · c7a243e · c7a243e
1 parent 8eec20d
commit c7a243e
Show file tree

Hide file tree

Showing 2 changed files with 4 additions and 3 deletions.
diff --git a/generation/2d_autoencoderkl/2d_autoencoderkl_tutorial.ipynb b/generation/2d_autoencoderkl/2d_autoencoderkl_tutorial.ipynb
@@ -46,7 +46,7 @@
    "id": "2caa73e1",
    "metadata": {},
    "source": [
-    "## Set up environment"
+    "## Setup environment"
    ]
   },
   {
@@ -79,6 +79,7 @@
     "import os\n",
     "import shutil\n",
     "import time\n",
+    "import tempfile\n",
     "import matplotlib.pyplot as plt\n",
     "import numpy as np\n",
     "import torch\n",

diff --git a/generation/3d_autoencoderkl/3d_autoencoderkl_tutorial.ipynb b/generation/3d_autoencoderkl/3d_autoencoderkl_tutorial.ipynb
@@ -74,6 +74,7 @@
    "source": [
     "import os\n",
     "import shutil\n",
+    "import tempfile\n",
     "import matplotlib.pyplot as plt\n",
     "import numpy as np\n",
     "import torch\n",
@@ -117,8 +118,7 @@
    ],
    "source": [
     "directory = os.environ.get(\"MONAI_DATA_DIRECTORY\")\n",
-    "root_dir = \"/tmp/tmp0_29f8jr\"\n",
-    "# root_dir = tempfile.mkdtemp() if directory is None else directory\n",
+    "root_dir = tempfile.mkdtemp() if directory is None else directory\n",
     "print(root_dir)"
    ]
   },