Fix fake transform docstrings

Author: ValerianRey

tests/unit/autojac/_transform/test_base.py

@@ -11,7 +11,7 @@
 
 class FakeTransform(Transform[_B, _C]):
     """
-    Fake ``Transform`` to test `required_keys` and `output_keys` when composing and conjuncting.
+    Fake ``Transform`` to test `check_keys` when composing and conjuncting.
     """
 
     def __init__(self, required_keys: set[Tensor], output_keys: set[Tensor]):
@@ -28,6 +28,7 @@ def __call__(self, input: _B) -> _C:
         return typing.cast(_C, output_dict)
 
     def check_keys(self, input_keys: set[Tensor]) -> set[Tensor]:
+        # Arbitrary requirement for testing purposes.
         if not input_keys == self._required_keys:
             raise RequirementError()
         return self._output_keys

tests/unit/autojac/_transform/test_stack.py

@@ -9,10 +9,7 @@
 
 
 class FakeGradientsTransform(Transform[EmptyTensorDict, Gradients]):
-    """
-    Transform that produces gradients filled with ones, for testing purposes. Note that it does the
-    same thing as Init, but it does not depend on Init.
-    """
+    """Transform that produces gradients filled with ones, for testing purposes."""
 
     def __init__(self, keys: Iterable[Tensor]):
         self.keys = set(keys)