earth2studio
earth2studio copied to clipboard
🐛[BUG]: SFNO/Makani missing some needed deps and fails on vanilla Pytorch container
Version
main
On which installation method(s) does this occur?
Pip, Source
Describe the issue
Present the SFNO model requires makani install.
This is inside nvcr.io/nvidia/pytorch:24.01-py3
The current deps of makani and also whats in e2studio are missing some items. Not sure what triggered this. I think something got updated in the Modulus version 0.6.0 that removed some deps that Makani assumed would be installed.
Packages needed to get added are:
pip install ruamel.yaml
pip install torch-harmonics
pip install tensorly
pip install tensorly-torch
```