f-dangel / curvlinops / 23204306094

91%

push

github

[REF] Rename `_model_func` to `_model_module`, add functional `_model_func` (#269)

* [REF] Rename _model_func to _model_module, add functional _model_func

Separate the two roles of _model_func:
- _model_module: the raw Module (for hooks backend, make_batch_* functions)
- _model_func: functional (params_dict, X) -> prediction via
  make_functional_call, used for predictions and shape inference

Migrate direct Module calls to the functional form where possible:
- _data_prediction_loss_gradient uses _model_func(self._params, X)
- Jacobian shape inference uses _model_func(self._params, x)
- KFAC/EKFAC FX backends use _model_func(params, x) instead of
  functional_call(self._model_module, params, (x,))

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

* [DOC] Add changelog entry for _model_func rename

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>

16 of 16 new or added lines in 10 files covered. (100.0%)

2340 of 2563 relevant lines covered (91.3%)

0.91 hits per line