Fix docstring of init.check_keys

ValerianRey · ValerianRey · commit 65bc0ae53b27 · 2025-03-31T17:11:34.000+02:00
diff --git a/src/torchjd/autojac/_transform/init.py b/src/torchjd/autojac/_transform/init.py
@@ -24,6 +24,6 @@ def __call__(self, input: EmptyTensorDict) -> Gradients:
     def check_keys(self, input_keys: set[Tensor]) -> set[Tensor]:
         if not input_keys == set():
             raise RequirementError(
-                f"The input_keys should be the empty set. Found input_keys" f"{input_keys}."
+                f"The input_keys should be the empty set. Found input_keys {input_keys}."
             )
         return self.values

Original file line number	Diff line number	Diff line change
`@@ -24,6 +24,6 @@ def __call__(self, input: EmptyTensorDict) -> Gradients:`
`24`	`24`	`def check_keys(self, input_keys: set[Tensor]) -> set[Tensor]:`
`25`	`25`	`if not input_keys == set():`
`26`	`26`	`raise RequirementError(`
`27`		`- f"The input_keys should be the empty set. Found input_keys" f"{input_keys}."`
	`27`	`+ f"The input_keys should be the empty set. Found input_keys {input_keys}."`
`28`	`28`	`)`
`29`	`29`	`return self.values`