From 6de5ca991457a2794dc32191641f5537fef594c3 Mon Sep 17 00:00:00 2001 From: Wessel Bruinsma Date: Fri, 10 Jan 2025 13:45:46 +0100 Subject: [PATCH] Remove random data --- docs/foundry/demo_hres_t0.ipynb | 16 ---------------- 1 file changed, 16 deletions(-) diff --git a/docs/foundry/demo_hres_t0.ipynb b/docs/foundry/demo_hres_t0.ipynb index fc2cac0..2edc40a 100644 --- 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",