NVIDIA / earth2studio

Open-source deep-learning framework for exploring, building and deploying AI weather/climate workflows.
https://nvidia.github.io/earth2studio/
Apache License 2.0
73 stars 23 forks source link

Pytest improvement, Bug Fixes and Documentation Update #73

Closed NickGeneva closed 2 months ago

NickGeneva commented 2 months ago

Earth2Studio Pull Request

Description

Cleaning up bugs:

Closes: https://github.com/NVIDIA/earth2studio/issues/72

Adds ability to control timeout of known remote stores using a environment variable for slow internet connections. Each remote store sets this is a different way... so theres not elegant solution. Also improved unit testing and fixed issue in S3 package.

Closes: https://github.com/NVIDIA/earth2studio/issues/71

Removed recursive check that appeared from copying the batch function decorator. Basically would call output_coords inside the batch coords decorator that would get it stuck. Simple fix. I still think we maybe want to change input_coords to a function that returns a new dictionary.

Closes: https://github.com/NVIDIA/earth2studio/issues/69

Adding testing page with pytest commands in the docs.

Attempts to fixxx https://github.com/NVIDIA/earth2studio/issues/68 by using scoped fixtures for some models. I dont know if this will fully fix it, but the model won't get created twice on the CPU now. Hopefully will speed up testing too.

Closes #75

Checklist

Dependencies

NickGeneva commented 2 months ago

/blossom-ci

NickGeneva commented 2 months ago

/blossom-ci

NickGeneva commented 2 months ago

/blossom-ci

NickGeneva commented 2 months ago

/blossom-ci