Document solar radiation library in README.md.

PiperOrigin-RevId: 595602930 Change-Id: I0b37369d622ab81c324d930f4ee2ebd798fb74dc
google-deepmind · Jan 4, 2024 · effffb2 · effffb2
1 parent e70b8ca
commit effffb2
Show file tree

Hide file tree

Showing 2 changed files with 4 additions and 1 deletion.
diff --git a/README.md b/README.md
@@ -60,6 +60,9 @@ The one-step implementation of GraphCast architecture, is provided in
     and all of the wrappers implement.
 *   `rollout.py`: Similar to `autoregressive.py` but used only at inference time
     using a python loop to produce longer, but non-differentiable trajectories.
+*   `solar_radiation.py`: Computes Top-Of-the-Atmosphere (TOA) incident solar
+    radiation compatible with ERA5. This is used as a forcing variable and thus
+    needs to be computed for target lead times in an operational setting.
 *   `typed_graph.py`: Definition of `TypedGraph`'s.
 *   `typed_graph_net.py`: Implementation of simple graph neural network
     building blocks defined over `TypedGraph`'s that can be combined to build

diff --git a/setup.py b/setup.py
@@ -35,14 +35,14 @@
         "colabtools",
         "dask",
         "dm-haiku",
-        "dm-tree",
         "jax",
         "jraph",
         "matplotlib",
         "numpy",
         "pandas",
         "rtree",
         "scipy",
+        "tree",
         "trimesh",
         "typing_extensions",
         "xarray",