f-dangel / curvlinops / 23111723348

91%

Build Type

push

github

Committed by web-flow

Commit Message

[REF] Introduce ParameterUsage dataclass for KFAC (#263)

* [REF] Introduce ParameterUsage dataclass for KFAC layer-parameter mapping

Unify KFAC's layer-to-parameter mapping into a single ParameterUsage
dataclass bundling operation type, parameter roles, and hyperparameters.
Rename mapping keys from "weight"/"bias" to "W"/"b" (matching
AffineLayerInfo notation), and change _mapping from dict to list.

Co-Authored-By: Claude Opus 4.6 (1M context) <noreply@anthropic.com>

* [DOC] Add PR link to changelog entry

Co-Authored-By: Claude Opus 4.6 (1M context) <noreply@anthropic.com>

* [REF] Extract _module_name_from_param helper

Co-Authored-By: Claude Opus 4.6 (1M context) <noreply@anthropic.com>

* [FIX] Address PR review: recurse=False, isinstance lookup, fix docstring

- Use recurse=False in named_parameters to avoid submodule params
- Use isinstance-based lookup for _module_info to support subclasses
- Fix stale docstring in EKFACComputer.compute()

Co-Authored-By: Claude Opus 4.6 (1M context) <noreply@anthropic.com>

---------

Co-authored-by: f-dangel <felix.dangel@concordia.ca>
Co-authored-by: Claude Opus 4.6 (1M context) <noreply@anthropic.com>

Run Details

83 of 83 new or added lines in 8 files covered. (100.0%)

15 existing lines in 7 files now uncovered.

2439 of 2667 relevant lines covered (91.45%)

0.91 hits per line