Improve RequirementError messages

Author: ValerianRey

src/torchjd/autojac/_transform/aggregate.py

@@ -51,7 +51,8 @@ def __call__(self, jacobian_matrices: JacobianMatrices) -> GradientVectors:
     def check_keys(self, input_keys: set[Tensor]) -> set[Tensor]:
         if not set(self.key_order) == input_keys:
             raise RequirementError(
-                f"The input_keys must match the key_order. Found {input_keys} and {self.key_order}"
+                f"The input_keys must match the key_order. Found input_keys {input_keys} and"
+                f"key_order {self.key_order}."
             )
         return input_keys
 

src/torchjd/autojac/_transform/diagonalize.py

@@ -31,7 +31,7 @@ def check_keys(self, input_keys: set[Tensor]) -> set[Tensor]:
         considered = set(self.considered)
         if not considered.issubset(input_keys):
             raise RequirementError(
-                f"The input_keys needs to be a super set of the considered keys. Found {input_keys} "
-                f"and {considered}"
+                f"The input_keys should be a super set of the considered keys. Found input_keys "
+                f"{input_keys} and considered keys {considered}."
             )
         return considered

src/torchjd/autojac/_transform/init.py

@@ -23,5 +23,5 @@ def __call__(self, input: EmptyTensorDict) -> Gradients:
 
     def check_keys(self, input_keys: set[Tensor]) -> set[Tensor]:
         if input_keys == set():
-            raise RequirementError(f"Init expects an empty set of input_keys. Found {input_keys}")
+            raise RequirementError(f"The input_keys should be the empty set. Found {input_keys}.")
         return self.values

src/torchjd/autojac/_transform/select.py

@@ -17,7 +17,7 @@ def __call__(self, tensor_dict: _A) -> _A:
     def check_keys(self, input_keys: set[Tensor]) -> set[Tensor]:
         if not self.keys.issubset(input_keys):
             raise RequirementError(
-                f"The input_keys needs to be a super set of the keys to select. Found {input_keys} "
-                f"and {self.keys}"
+                f"The input_keys should be a super set of the keys to select. Found input_keys "
+                f"{input_keys} and keys to select {self.keys}."
             )
         return self.keys