Adding GEFS forecast data source (#112)

* Adding GEFS data source * Adding tests for GEFS * Feedback * Fixes
NVIDIA · Aug 12, 2024 · 97f80d4 · 97f80d4
1 parent a95eab5
commit 97f80d4
Show file tree

Hide file tree

Showing 8 changed files with 850 additions and 3 deletions.
diff --git a/CHANGELOG.md b/CHANGELOG.md
@@ -11,7 +11,8 @@ and this project adheres to [Semantic Versioning](https://semver.org/spec/v2.0.0
 ### Added
 
 - Forecast datasource API
-- GFS Forecast datasource
+- GFS forecast datasource
+- GEFS forecast datasource
 
 ### Changed
 

diff --git a/docs/modules/datasources.rst b/docs/modules/datasources.rst
@@ -51,6 +51,7 @@ Typically used in intercomparison workflows.
    :template: datasource.rst
 
    data.GFS_FX
+   data.GEFS_FX
 
 Functions
 ~~~~~~~~~

diff --git a/earth2studio/data/__init__.py b/earth2studio/data/__init__.py
@@ -17,6 +17,7 @@
 from .arco import ARCO
 from .base import DataSource
 from .cds import CDS
+from .gefs import GEFS_FX
 from .gfs import GFS, GFS_FX
 from .hrrr import HRRR
 from .ifs import IFS