How to wrap my trained model so that I can call it from earth2studio?

[choutilin]

Dear Sir/Madam,

Thank you for providing this repo, I've used it to train my own model checkpoints.
I can run inference using makani, but I'd also like to try using the NVIDIA/earth2studio framework to run inference.
How should I wrap my model? I see there's a makani/models/model_package.py for earth2mip. Will it also be compatible with earth2studio? (Actually, is earth2mip deprecated by now?)

Thanks again.
Ti-Lin Chou