NVIDIA-NeMo / NVIDIA-NeMo/Automodel
Expand ty type-checking coverage in AutoModel
Open
@akoumpa is already working on this.
Since May 12, 2026.
coverage
enhancement
- Dominant language
- Python
- Stars
- 963
- Forks
- 318
- Avg merge
- 3d 20h
- Merged PRs (30d)
- 143
Description
Feature Request
Expand AutoModel's ty type-checking coverage beyond the current Phase 1 modules.
ty is already configured in pyproject.toml and enforced in CI, but strict checks are currently active only for attention, config, launcher, loggers, and optim modules. We should gradually enable the existing Phase 2/Phase 3 targets and fix the resulting type issues.
Proposed Scope
- Enable the next
tool.ty.overridestarget group inpyproject.toml. - Fix reported
tyerrors without broadening ignores unnecessarily. - Keep CI
type_checkingpassing. - Preserve the phased rollout so future modules can be added incrementally.
Acceptance Criteria
- Additional AutoModel modules are checked by
tyin CI. - No new broad unresolved-import or unresolved-reference suppressions are added unless justified.
- The active coverage list in
pyproject.tomlreflects the expanded scope.
Contributor guide
First steps
- Read the whole issue, then the project's contributing guide.
- Comment on the issue to say you are picking it up — it saves two people doing the same work.
- Fork the repository and make your change on a branch.
- Open a pull request that references the issue number.
Assessment
This issue has not been assessed yet.