Skip to content
New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? Sign in to your account

Bnb/masked fwp #231

Draft
wants to merge 385 commits into
base: main
Choose a base branch
from
Draft

Bnb/masked fwp #231

wants to merge 385 commits into from

Conversation

bnb32
Copy link
Collaborator

@bnb32 bnb32 commented Sep 25, 2024

This adds the option to skip chunks in the forward pass routine if all coordinates covered by that chunk are masked. The mask is provided as an additional variable through file_paths, where mask is 1 / True if the coordinate should not be included in the forward pass. An example which masks conus except for areas around observation locations is shown below.

Mask:
conus_val_mask

Collected forward pass output:
conus_fwp_masked

bnb32 added 30 commits July 23, 2024 15:32
…tory meta class. moved bias correct from ForwardPass, done for each chunk, to ForwardPassStrategy init.
…dler which invoked log interp. Thus the log interp methods from era_downloader can be removed and LogLinInterpolator can also be removed.
…old Feature class since we now have `parse_feature` method.
… returns type specific handler based on file type
…rent time indices, like for monthly files. also added catch in cacher for weird dimensions that we shouldnt be writing to h5
…t features. now just repeat over time dimension in the forward pass padding method. this is much more performant for time independent features. this doesnt solve slow down issues with time dependent exo features like sza though.
…g. u_30m from u_10m and u_100m, with u pressure level array
… Added shape checks in ``test_access.py``
…dataset() which will default to a dask array manager if chunks is specified and load into memory as numpy arrays in chunks is None.
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment
Labels
None yet
Projects
None yet
Development

Successfully merging this pull request may close these issues.

3 participants