We read every piece of feedback, and take your input very seriously.
To see all available qualifiers, see our documentation.
1 parent 2dc2433 commit d160b7bCopy full SHA for d160b7b
src/torchjd/autojac/_transform/init.py
@@ -22,6 +22,6 @@ def __call__(self, input: EmptyTensorDict) -> Gradients:
22
return Gradients({value: torch.ones_like(value) for value in self.values})
23
24
def check_keys(self, input_keys: set[Tensor]) -> set[Tensor]:
25
- if input_keys == set():
+ if not input_keys == set():
26
raise RequirementError(f"The input_keys should be the empty set. Found {input_keys}.")
27
return self.values
0 commit comments