Fix docstring formatting

Author: ValerianRey

src/torchjd/aggregation/__init__.py

@@ -1,7 +1,8 @@
 r"""
 When doing Jacobian descent, the Jacobian matrix has to be aggregated into a vector to store in the
 ``.grad`` fields of the model parameters. The
-The :class:`~torchjd.aggregation._aggregator_bases.Aggregator` is responsible for these aggregations.
+The :class:`~torchjd.aggregation._aggregator_bases.Aggregator` is responsible for these
+aggregations.
 
 When using the :doc:`autogram <../autogram/index>` engine, we rather need to extract a vector
 of weights from the Gramian of the Jacobian. The

src/torchjd/autogram/_edge_registry.py

@@ -5,8 +5,8 @@
 
 class EdgeRegistry:
     """
-    Tracks `GradientEdge`s and provides a way to efficiently compute a minimally sufficient subset of
-    leaf edges that are reachable from some given `GradientEdge`s.
+    Tracks `GradientEdge`s and provides a way to efficiently compute a minimally sufficient subset
+    of leaf edges that are reachable from some given `GradientEdge`s.
     """
 
     def __init__(self) -> None:

src/torchjd/autogram/_module_hook_manager.py

@@ -103,9 +103,9 @@ class JacobianAccumulator(torch.autograd.Function):
             """
             Autograd function that accumulates Jacobian Gramians during the first backward pass.
 
-            Acts as identity on forward pass. During the autogram algorithm,
-            computes the Jacobian of outputs w.r.t. module parameters and feeds it to the gramian
-            accumulator. Uses a toggle mechanism to activate only during the Gramian accumulation phase.
+            Acts as identity on forward pass. During the autogram algorithm, computes the Jacobian
+            of outputs w.r.t. module parameters and feeds it to the gramian accumulator. Uses a
+            toggle mechanism to activate only during the Gramian accumulation phase.
             """
 
             @staticmethod