From ba42e439b9f4d597d22c2e4c067a224b07ef7eb2 Mon Sep 17 00:00:00 2001 From: Wessel Bruinsma Date: Fri, 13 Sep 2024 15:02:59 +0200 Subject: [PATCH 1/2] Remove redundant raises --- aurora/model/aurora.py | 3 --- 1 file changed, 3 deletions(-) diff --git a/aurora/model/aurora.py b/aurora/model/aurora.py index 9fae270..4c68c6b 100644 --- a/aurora/model/aurora.py +++ b/aurora/model/aurora.py @@ -158,9 +158,6 @@ def forward(self, batch: Batch) -> Batch: Args: batch (:class:`Batch`): Batch to run the model on. - Raises: - ValueError: If no metric is provided. - Returns: :class:`Batch`: Prediction for the batch. """ From 109600fe5d8c3665db29e0cb0cde7794c4ec5bc4 Mon Sep 17 00:00:00 2001 From: Wessel Bruinsma Date: Fri, 13 Sep 2024 15:03:08 +0200 Subject: [PATCH 2/2] Add strict=False when adding new variables --- docs/finetuning.md | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/docs/finetuning.md b/docs/finetuning.md index 8bdda7b..266f894 100644 --- a/docs/finetuning.md +++ b/docs/finetuning.md @@ -53,7 +53,7 @@ model = Aurora( static_vars=("lsm", "z", "slt", "new_static_var"), atmos_vars=("z", "u", "v", "t", "q", "new_atmos_var"), ) -model.load_checkpoint("microsoft/aurora", "aurora-0.25-pretrained.ckpt") +model.load_checkpoint("microsoft/aurora", "aurora-0.25-pretrained.ckpt", strict=False) # Normalisation means: locations["new_surf_var"] = 0.0