Skip to content

Weather utilities module#36

Merged
dsj976 merged 38 commits intomainfrom
implement/weather-utils
Apr 1, 2026
Merged

Weather utilities module#36
dsj976 merged 38 commits intomainfrom
implement/weather-utils

Conversation

@dsj976
Copy link
Copy Markdown
Collaborator

@dsj976 dsj976 commented Mar 25, 2026 •
edited
Loading

This PR implements a new module to perform operations typically found in data-driven weather forecasting, including:

  • Computing the RMSE score
  • Computing the MAE score
  • Computing climatology
  • Computing the Continuous Ranked Probability Score (CRPS) for probabilistic forecasts
  • Computing the Anomaly Correlation Coefficient
  • Computing the energy spectrum

Two additional dependencies have been included in order to achieve this:

dsj976 added 30 commits March 11, 2026 15:45
@dsj976
fix: compute_climatology() now correctly handling leap years and star…
f0b32fb 
…t months different from Jan
@dsj976
update pydmd dependency URL to the official repository
6d776c2
@dsj976
feat: add compute_rmse function and corresponding tests for RMSE calc…
ea47a5a 
…ulation
@dsj976
Merge branch 'fix-compute-climatology' into implement/weather-utils
e959ca5
@dsj976
feat: implement compute_climatology function for climatological forec…
bf97437 
…asts
@dsj976
do not trigger Dask computation in compute_rmse()
985eb51
@dsj976
feat: add compute_energy_spectrum function using Weatherbench2 for zo…
65b73a4 
…nal energy spectrum calculations
@dsj976
refactor: improve DataGenerator initialization and restructure RMSE t…
9410d6a 
…est with fixture
@dsj976
refactor: simplify latitude weighting in compute_rmse function
28179ac
@dsj976
reduce test dataset size in test_weather_utils
b5e9bdf
@dsj976
check for expected rmse in test_compute_rmse
0ccf572
@dsj976
drop time in climatology before new assignment to avoid deprecation w…
6dce95b 
…arning
@dsj976
test for compute_climatology function
1d09011
@dsj976
add test for compute_energy_spectrum function
060bb25
@dsj976
ignore PendingDeprecationWarning from weatherbench2 in pytest
61b3a94
@dsj976
update climatology to be returned as a function of day of year and ho…
8662bdf 
…ur of day
@dsj976
add expand_time_climatology function to convert dayofyear and hour di…
d6b0256 
…mensions into a single time dimension
@dsj976
update compute_climatology: remove months parameter and do not drop 2…
4bda48d 
…9 Feb
@dsj976
add pytest-dependency to development dependencies
139967a
@dsj976
fix bug in time vector generation in expand_time_climatology()
0f94491
@dsj976
add test for expand_time_climatology function in test_weather_utils.py
065afdb
@dsj976
feat: enhance compute_climatology with optional smoothing window
d486b76
@dsj976
docs: update compute_climatology notes on smoothing performance and d…
e68c904 
…isk usage
@dsj976
improve compute_rmse function to raise error in case of empty array
b0c5559
@dsj976
feat: add compute_crps_gaussian function for probabilistic forecast e…
6c18344 
…valuation
@dsj976
fix: update compute_crps_gaussian to handle lat_weighting and dims ch…
bfdc12e 
…ecks correctly
@dsj976
add test for compute_crps_gaussian()
87011b8
@dsj976
add compute_acc function for anomaly correlation calculation
3f723af
@dsj976
add docstring and input checks to compute_acc function
496e7de
@dsj976
fix bug in expand_time_climatology()
2d9eed6
@dsj976 dsj976 requested a review from robertvava March 25, 2026 14:40
@robertvava
Copy link
Copy Markdown
Collaborator

robertvava commented Mar 29, 2026

Looks good to go! Will add some suggestions in a PR. For now, this can be merged!

@dsj976 dsj976 merged commit d8218c8 into main Apr 1, 2026
5 checks passed
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment

Reviewers

@robertvava robertvava robertvava approved these changes

Assignees

No one assigned

Labels

None yet

Projects

None yet

Milestone

No milestone

Development

Successfully merging this pull request may close these issues.

2 participants

@dsj976 @robertvava