Remove non-blocking IO writes

[jleinonen]

Earth2Studio Pull Request

Description

This PR removes non_blocking=True from .to() statements that were causing hard-to-trace IO issues.

Checklist

• [x] I am familiar with the Contributing Guidelines.
• [x] New or existing tests cover these changes.
• [x] The documentation is up to date with these changes.
• [ ] The CHANGELOG.md is up to date with these changes.
• [ ] An issue is linked to this pull request.

Dependencies

[NickGeneva]

/blossom-ci