Skip to content

Commit 4d774fa

Browse files
committed
update: test_no_gradients
1 parent 2c7b606 commit 4d774fa

File tree

1 file changed

+1
-1
lines changed

1 file changed

+1
-1
lines changed

tests/test_gradients.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -9,7 +9,7 @@
99

1010
@pytest.mark.parametrize('optimizer_name', [*VALID_OPTIMIZER_NAMES, 'lookahead'])
1111
def test_no_gradients(optimizer_name):
12-
if optimizer_name == 'lomo':
12+
if optimizer_name in {'lomo', 'adammini'}:
1313
pytest.skip(f'skip {optimizer_name} optimizer.')
1414

1515
p1 = simple_parameter(require_grad=True)

0 commit comments

Comments
 (0)