Makes a couple of fixes for DLIO use case

Author: ilumsden

2024-RIKEN-AWS/JupyterNotebook/tutorial/notebook/dyad_dlio.ipynb

@@ -151,7 +151,7 @@
    "outputs": [],
    "source": [
     "kvs_namespace = \"dyad\"\n",
-    "managed_directory = \"/tmp/dyad_data\"\n",
+    "managed_directory = \"./dyad_data\"\n",
     "dtl_mode = \"UCX\" # We currently only support UCX, so do not change this"
    ]
   },
@@ -170,7 +170,7 @@
    "metadata": {},
    "outputs": [],
    "source": [
-    "initial_data_directory = \"/tmp/dlio_data\""
+    "initial_data_directory = \"./dlio_data\""
    ]
   },
   {
@@ -190,8 +190,8 @@
    "source": [
     "workers_per_node = 8\n",
     "dyad_install_prefix = \"/usr/local\"\n",
-    "num_nodes = !flux hostlist -c\n",
-    "dlio_extensions_dir = !$HOME/flux-tutorial-2024/dlio_extensions\n",
+    "num_nodes = 4\n",
+    "dlio_extensions_dir = \"/home/jovyan/flux-tutorial-2024/dlio_extensions\"\n",
     "workload = \"dyad_unet3d_small\""
    ]
   },
@@ -210,7 +210,7 @@
    "metadata": {},
    "outputs": [],
    "source": [
-    "env_file = \"\"\"\n",
+    "env_file = f\"\"\"\n",
     "DYAD_KVS_NAMESPACE={kvs_namespace}\n",
     "DYAD_DTL_MODE={dtl_mode}\n",
     "DYAD_PATH={managed_directory}\n",