f-dangel / curvlinops / 11667562849

90%

push

github

[REF] Replicate base class, but inherit from PyTorch linear operator (#142)

* [ADD] Minimal linear operator interface for PyTorch

* [FIX] Linters

* [ADD] Replicate current base class but inherit from `PyTorchLinearOperator`

* [DEL] Unused import

* [DEL] Remove unused import

* [REF] Address minor comments

* [FIX] Type hint in `_matmat_batch`

* [ADD] Test `gradient_and_loss` function from `CurvatureLinearOperator`

65 of 116 new or added lines in 3 files covered. (56.03%)

1532 of 1708 relevant lines covered (89.7%)

0.9 hits per line