f-dangel / curvlinops / 22831429882

91%

main: 92%

Build Type

Pull #220

github

Committed by f-dangel

Commit Message

[FIX] Update type annotations and docstrings for tuple parameter convention

Fix return type annotations in make_functional_call,
make_functional_model_and_loss, and make_functional_flattened_model_and_loss
to reflect the new (params, X) signature. Add MutableMapping to X type hints
in inner functions. Update _mp docstrings to describe tuple-based matrix args.

Co-Authored-By: Claude Opus 4.6 <noreply@anthropic.com>

Pull Request Pull Request #220: [REF] Stylistic improvements in `torch.func` functionality

Coverage Stats

34 of 34 new or added lines in 7 files covered. (100.0%)

2 existing lines in 2 files now uncovered.

2207 of 2418 relevant lines covered (91.27%)

0.91 hits per line