Add notebook for HRES 0.1 deg predictions (#9)

* Fix link to ERA5 notebook. * Added util.py file for notebooks * Added notebook for 0.1 model * Typo * Typo. * Fix util.py * Update pre-commit * Fix docstrings. * Adjust filenames * Fixed imports * Added function to download static vars from RDA * Added function to download static vars from RDA * Updated 0.1deg notebook * Fix link to ERA5 notebook in HRES T0 notebook * Clean up 0.1 notebook * Add regridding * RM * v2 * Add notebook * Fix * Rm interpolate function * Update notebook * Fix * Revert changes to HRES T0 notebook * Simplify data loading * Run notebook * Add warning about adjustment of statistics --------- Co-authored-by: Wessel Bruinsma <[email protected]>
microsoft · Sep 15, 2024 · a3a0c44 · a3a0c44
1 parent 00a1a7a
commit a3a0c44
Show file tree

Hide file tree

Showing 3 changed files with 298 additions and 0 deletions.
diff --git a/.pre-commit-config.yaml b/.pre-commit-config.yaml
@@ -42,3 +42,5 @@ repos:
     rev: "v1.11.0"
     hooks:
     - id: mypy
+      additional_dependencies:
+          - "types-requests"
diff --git a/aurora/model/aurora.py b/aurora/model/aurora.py
@@ -2,6 +2,7 @@
 
 import contextlib
 import dataclasses
+import warnings
 from datetime import timedelta
 from functools import partial
 from typing import Optional
@@ -112,6 +113,14 @@ def __init__(
         self.surf_stats = surf_stats or dict()
         self.autocast = autocast
 
+        if self.surf_stats:
+            warnings.warn(
+                f"The normalisation statics for the following surface-level variables are manually "
+                f"adjusted: {', '.join(sorted(self.surf_stats.keys()))}. "
+                f"Please ensure that this is right!",
+                stacklevel=2,
+            )
+
         self.encoder = Perceiver3DEncoder(
             surf_vars=surf_vars,
             static_vars=static_vars,

diff --git a/docs/example_hres_0.1.ipynb b/docs/example_hres_0.1.ipynb