Let the demo depend on env vars

microsoft · Jan 10, 2025 · 842dbc7 · 842dbc7
1 parent 27321cd
commit 842dbc7
Showing 1 changed file with 4 additions and 3 deletions.
diff --git a/docs/foundry/demo_hres_t0.ipynb b/docs/foundry/demo_hres_t0.ipynb
@@ -41,6 +41,7 @@
    "metadata": {},
    "outputs": [],
    "source": [
+    "import os\n",
     "from datetime import datetime\n",
     "\n",
     "import torch\n",
@@ -49,10 +50,10 @@
     "from aurora.foundry import BlobStorageChannel, FoundryClient, submit\n",
     "\n",
     "foundry_client = FoundryClient(\n",
-    "    endpoint=\"https://endpoint/\",\n",
-    "    token=\"TOKEN\",\n",
+    "    endpoint=os.environ[\"FOUNDRY_ENDPOINT\"],\n",
+    "    token=os.environ[\"FOUNDRY_TOKEN\"],\n",
     ")\n",
-    "channel = BlobStorageChannel(\"https://my.blob.core.windows.net/container/folder?<SAS_TOKEN>\")\n",
+    "channel = BlobStorageChannel(os.environ[\"BLOB_URL_WITH_SAS\"])\n",
     "\n",
     "initial_condition = Batch(\n",
     "    surf_vars={k: torch.randn(1, 2, 17, 32) for k in (\"2t\", \"10u\", \"10v\", \"msl\")},\n",