Fix docstring and error catching in test

curvlinops/kfac.py

@@ -243,7 +243,11 @@ def __init__(
         Raises:
             RuntimeError: If the check for deterministic behavior fails.
             ValueError: If the loss function is not supported.
+            ValueError: If the Fisher type is not supported.
+            ValueError: If the KFAC approximation type is not supported.
             ValueError: If ``fisher_type != FisherType.MC`` and ``mc_samples != 1``.
+            NotImplementedError: If ``correct_eigenvalues`` and ``fisher_type ==
+                FisherType.FORWARD_ONLY``.
             ValueError: If ``X`` is not a tensor and ``batch_size_fn`` is not specified.
         """
         if not isinstance(loss_func, self._SUPPORTED_LOSSES):

test/test_kfac.py

@@ -512,7 +512,7 @@ def test_multi_dim_output(
     # KFAC for deep linear network with 4d input and output
     params = list(model.parameters())
     context = (
-        raises(ValueError, match="eigenvalues")
+        raises(NotImplementedError, match="eigenvalues")
         if correct_eigenvalues and fisher_type == FisherType.FORWARD_ONLY
         else nullcontext()
     )  # EKFAC for FOOF is currently not supported