Skip to content

Commit 0a49543

Browse files
committed
Improve docstring test_composition_check_keys
1 parent 25a3980 commit 0a49543

1 file changed

Lines changed: 1 addition & 1 deletion

File tree

tests/unit/autojac/_transform/test_base.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -36,7 +36,7 @@ def check_keys(self, input_keys: set[Tensor]) -> set[Tensor]:
3636
def test_composition_check_keys():
3737
"""
3838
Tests that `check_keys` works correctly for a composition of transforms: the inner transform's
39-
`output_keys` has to match with the outer transform's `required_keys`.
39+
`output_keys` has to satisfy the outer transform's requirements.
4040
"""
4141

4242
a1 = torch.randn([2])

0 commit comments

Comments
 (0)