Fix black

f-dangel · Nov 7, 2023 · e4e6b75 · e4e6b75
1 parent b67adc0
commit e4e6b75
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/curvlinops/kfac.py b/curvlinops/kfac.py
@@ -262,12 +262,12 @@ def _compute_kfac(self):
 
  def _compute_loss_and_backward(self, output: Tensor, y: Tensor):
  """Compute the loss and the backward pass(es) required for KFAC.
- 
+
  Args:
  output: The model's prediction
  :math:`\{f_\mathbf{\theta}(\mathbf{x}_n)\}_{n=1}^N`.
  y: The labels :math:`\{\mathbf{y}_n\}_{n=1}^N`.
- 
+
  Raises:
  NotImplementedError: If ``fisher_type == 'type-2'`` and
  ``isinstance(self._loss_func, CrossEntropyLoss)``.