Skip to content

Commit 5f48574

Browse files
committed
Fix type of raise in test_stack_check_keys
1 parent ccdc14f commit 5f48574

1 file changed

Lines changed: 2 additions & 1 deletion

File tree

tests/unit/autojac/_transform/test_interactions.py

Lines changed: 2 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -12,6 +12,7 @@
1212
Init,
1313
Jac,
1414
Jacobians,
15+
RequirementError,
1516
Select,
1617
Stack,
1718
TensorDict,
@@ -269,5 +270,5 @@ def test_stack_check_keys():
269270

270271
assert output_keys == {a}
271272

272-
with raises(ValueError):
273+
with raises(RequirementError):
273274
Stack([grad1, grad3]).check_keys({y1, y2})

0 commit comments

Comments
 (0)