From effffb202d47b46f4456d934548b890dfb0df164 Mon Sep 17 00:00:00 2001
From: Octavian Voicu <octav@google.com>
Date: Wed, 3 Jan 2024 23:02:18 -0800
Subject: [PATCH] Document solar radiation library in README.md.

PiperOrigin-RevId: 595602930
Change-Id: I0b37369d622ab81c324d930f4ee2ebd798fb74dc
---
 README.md | 3 +++
 setup.py  | 2 +-
 2 files changed, 4 insertions(+), 1 deletion(-)

diff --git a/README.md b/README.md
index efa89fc..aaa8274 100644
--- 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
index 692f951..04f449b 100644
--- a/setup.py
+++ b/setup.py
@@ -35,7 +35,6 @@
         "colabtools",
         "dask",
         "dm-haiku",
-        "dm-tree",
         "jax",
         "jraph",
         "matplotlib",
@@ -43,6 +42,7 @@
         "pandas",
         "rtree",
         "scipy",
+        "tree",
         "trimesh",
         "typing_extensions",
         "xarray",