Remove random data

microsoft · Jan 10, 2025 · 6de5ca9 · 6de5ca9
1 parent fd6ef69
commit 6de5ca9
Showing 1 changed file with 0 additions and 16 deletions.
diff --git a/docs/foundry/demo_hres_t0.ipynb b/docs/foundry/demo_hres_t0.ipynb
@@ -42,11 +42,7 @@
    "outputs": [],
    "source": [
     "import os\n",
-    "from datetime import datetime\n",
     "\n",
-    "import torch\n",
-    "\n",
-    "from aurora import Batch, Metadata\n",
     "from aurora.foundry import BlobStorageChannel, FoundryClient, submit\n",
     "\n",
     "foundry_client = FoundryClient(\n",
@@ -55,18 +51,6 @@
     ")\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",
-    "    static_vars={k: torch.randn(17, 32) for k in (\"lsm\", \"z\", \"slt\")},\n",
-    "    atmos_vars={k: torch.randn(1, 2, 4, 17, 32) for k in (\"z\", \"u\", \"v\", \"t\", \"q\")},\n",
-    "    metadata=Metadata(\n",
-    "        lat=torch.linspace(90, -90, 17),\n",
-    "        lon=torch.linspace(0, 360, 32 + 1)[:-1],\n",
-    "        time=(datetime(2020, 6, 1, 12, 0),),\n",
-    "        atmos_levels=(100, 250, 500, 850),\n",
-    "    ),\n",
-    ")\n",
-    "\n",
     "predictions = list(\n",
     "    submit(\n",
     "        initial_condition,\n",