physicsnemo
physicsnemo copied to clipboard
Docs improvement for `Module.from_checkpoint`
PhysicsNeMo Pull Request
Description
This PR improves the docstring for the method Module.from_checkpoint with added details regarding the mechanism to infer the class, as well as recommended usage.
Note: This PR has a companion PR in the documentation repo, and they should be merged together.
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.
- [x] The CHANGELOG.md is up to date with these changes.
- [x] An issue is linked to this pull request.