earth2studio
earth2studio copied to clipboard
🐛[BUG]: Graphcast coverage needs improvement
Version
main
On which installation method(s) does this occur?
No response
Describe the issue
earth2studio/models/px/graphcast_small.py 232 48 38 4 81% 39-49, 363-438, 530-531, 628, 630->637, 648
Need to update the package test to create the iterator and test the forward pass using that than just the calls. The mocks needed because of the Jax part causes a pretty big gap it seems.