Skip to content
Open
Show file tree
Hide file tree
Changes from 47 commits
Commits
Show all changes
48 commits
Select commit Hold shift + click to select a range
3ac2689
tempest extremes diagnostic model
mariusaurus Sep 15, 2025
d9653ab
enable setting fcn3 random seed
dallasfoster Sep 19, 2025
bf1b6a5
add proper noise handling for fcn3
dallasfoster Sep 19, 2025
6e601a2
fix linting and test issues
dallasfoster Sep 19, 2025
88fab22
update lockfile
dallasfoster Sep 19, 2025
591895c
update changelog
dallasfoster Sep 19, 2025
62b146b
Merge branch 'main' into dallasf/fcn3_random_noise_fix
NickGeneva Sep 19, 2025
e6ec7b8
Doc string
NickGeneva Sep 19, 2025
ffa348e
move seed initialization and fix dxwrapper tests
dallasfoster Sep 19, 2025
671ce23
Merge branch 'dallasf/fcn3_random_noise_fix' of github.com:dallasfost…
dallasfoster Sep 19, 2025
894b322
Merge branch 'tempest_extremes' into dallasf/fcn3_random_noise_fix
mariusaurus Sep 22, 2025
6bc7089
Merge branch 'main' into tempest_extremes
mariusaurus Sep 23, 2025
dcf839f
Merge branch 'tempest_extremes' of github.com:mariusaurus/earth2studi…
mariusaurus Sep 23, 2025
2fe8c30
error message
mariusaurus Sep 25, 2025
e45a248
Merge branch 'tempest_extremes' of github.com:mariusaurus/earth2studi…
mariusaurus Sep 25, 2025
05162e0
testing if TE is available and works
mariusaurus Sep 25, 2025
904203c
started working on support for batch sizes >1, currently works for bs 1
mariusaurus Sep 25, 2025
54b5e7d
halfway to larger batch support
mariusaurus Sep 25, 2025
1ef8acb
enabling TE for batch sizes of >1. async version seems to work as wel…
mariusaurus Sep 26, 2025
35f2af5
option to pass file names to TE connector
mariusaurus Sep 29, 2025
08c61a3
array equal test
mariusaurus Sep 29, 2025
a2e0354
Merge branch 'main' of github.com:NVIDIA/earth2studio into tempest_ex…
mariusaurus Oct 2, 2025
03da9f8
Merge branch 'main' of github.com:NVIDIA/earth2studio
mariusaurus Oct 2, 2025
9043cd4
Merge branch 'main' into tempest_extremes
mariusaurus Oct 2, 2025
7e7699d
first stable try
mariusaurus Oct 7, 2025
ee1863f
support for per-member parallel execution and lets user controll max …
mariusaurus Oct 7, 2025
7ea34fe
precommit hooks
mariusaurus Oct 7, 2025
d86102b
vibe-coded some tests, need to be hand-tested and selected
mariusaurus Oct 7, 2025
5f598a0
vibe-coded some tests, need to be hand-tested and selected
mariusaurus Oct 7, 2025
6b03cb7
Merge branch 'main' of github.com:NVIDIA/earth2studio
mariusaurus Oct 7, 2025
c4276ae
Merge branch 'main' into tempest_extremes
mariusaurus Oct 7, 2025
97f4c2f
passing all pre-commit tests, still need to sub-select tests as there…
mariusaurus Oct 7, 2025
85c4df3
subselected tests
mariusaurus Oct 7, 2025
a4c41c4
throwing an error in case cleanup is not called before object goes ou…
mariusaurus Oct 8, 2025
7caf87b
Merge branch 'main' of github.com:NVIDIA/earth2studio
mariusaurus Oct 8, 2025
5e3b7e3
Merge branch 'main' into tempest_extremes
mariusaurus Oct 8, 2025
f3cad7d
install doc
mariusaurus Oct 8, 2025
1732d0b
Merge branch 'tempest_extremes' of github.com:mariusaurus/earth2studi…
mariusaurus Oct 8, 2025
b73acdd
custom depenmdency failure message for TE
mariusaurus Oct 8, 2025
70d9b33
moved tensor tiling and concatenation to utils
mariusaurus Oct 8, 2025
dfe5494
Merge branch 'main' of github.com:NVIDIA/earth2studio
mariusaurus Oct 24, 2025
402b837
Merge branch 'main' of github.com:NVIDIA/earth2studio
mariusaurus Oct 30, 2025
c1aed2d
Merge branch 'main' of github.com:mariusaurus/earth2studio
mariusaurus Oct 30, 2025
eee67bb
Merge branch 'main' into tempest_extremes
mariusaurus Oct 30, 2025
b368325
Merge branch 'tempest_extremes' of github.com:mariusaurus/earth2studi…
mariusaurus Oct 30, 2025
97fe56f
addressed greptile comments
mariusaurus Oct 30, 2025
198dfe5
time import
mariusaurus Oct 30, 2025
e87c47d
comma
mariusaurus Oct 30, 2025
File filter

Filter by extension

Filter by extension

Conversations
Failed to load comments.
Loading
Jump to
Jump to file
Failed to load files.
Loading
Diff view
Diff view
14 changes: 12 additions & 2 deletions docs/userguide/about/install.md
Original file line number Diff line number Diff line change
Expand Up @@ -455,8 +455,8 @@ uv add earth2studio --extra corrdiff
::::
:::::
:::::{tab-item} Cyclone Trackers
Notes: Additional dependencies for all cyclone tracking models. Only Python 3.12 and
below support.
Notes: Additional dependencies for cyclone tracking models `TCTrackerVitart` and `TCTrackerWuDuan`.
Only Python 3.12 and below support.

::::{tab-set}
:::{tab-item} uv
Expand All @@ -472,6 +472,16 @@ uv pip install earth2studio --extra cyclone
pip install earth2studio[cyclone]
```

`TempestExtremes` is not provided as a Python library and must be installed
separately by the user. Installation instructions can be found on the
[TempestExtremes GitHub page](https://github.com/ClimateGlobalChange/tempestextremes?tab=readme-ov-file#installation-via-cmake-recommended).

By default, `TempestExtremes` binaries are installed to the bin folder of the
`TempestExtremes` repository. Users can specify paths to executables by passing
the detect and stitch commands to the initialisation of the `TempestExtremes`
class in earth2studio. Examples for both commands are provided in the docstring
of this class.

:::
::::
:::::
Expand Down
7 changes: 6 additions & 1 deletion earth2studio/models/dx/__init__.py
Original file line number Diff line number Diff line change
Expand Up @@ -38,15 +38,20 @@
TCTrackerVitart,
TCTrackerWuDuan,
)
from earth2studio.models.dx.tempest_extremes import (
AsyncTempestExtremes,
TempestExtremes,
)
from earth2studio.models.dx.wind_gust import WindgustAFNO # noqa

__all__ = [
"ClimateNet",
"AsyncTempestExtremes" "ClimateNet",
Copy link

Choose a reason for hiding this comment

The reason will be displayed to describe this comment to others. Learn more.

syntax: Missing comma between "AsyncTempestExtremes" and "ClimateNet" causes string concatenation instead of separate list items

Suggested change
"AsyncTempestExtremes" "ClimateNet",
"AsyncTempestExtremes",
"ClimateNet",

"CorrDiff",
"CorrDiffTaiwan",
"PrecipitationAFNO",
"PrecipitationAFNOv2",
"SolarRadiationAFNO1H",
"SolarRadiationAFNO6H",
"TempestExtremes",
"WindgustAFNO",
]
Loading