Skip to content

Commit d24fee1

Browse files
committed
uniformize error message of init.check_keys with others
1 parent 59dbdf3 commit d24fee1

1 file changed

Lines changed: 3 additions & 1 deletion

File tree

  • src/torchjd/autojac/_transform

src/torchjd/autojac/_transform/init.py

Lines changed: 3 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -23,5 +23,7 @@ def __call__(self, input: EmptyTensorDict) -> Gradients:
2323

2424
def check_keys(self, input_keys: set[Tensor]) -> set[Tensor]:
2525
if not input_keys == set():
26-
raise RequirementError(f"The input_keys should be the empty set. Found {input_keys}.")
26+
raise RequirementError(
27+
f"The input_keys should be the empty set. Found input_keys" f"{input_keys}."
28+
)
2729
return self.values

0 commit comments

Comments
 (0)