Skip to content

Commit beb84d4

Browse files
committed
update: test_bf16_gradient
1 parent 4f497d9 commit beb84d4

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
@@ -113,7 +113,7 @@ def test_sparse_supported(sparse_optimizer):
113113

114114
@pytest.mark.parametrize('optimizer_name', VALID_OPTIMIZER_NAMES)
115115
def test_bf16_gradient(optimizer_name):
116-
if optimizer_name in ('shampoo', 'lomo', 'bsam'):
116+
if optimizer_name in ('shampoo', 'lomo', 'bsam', 'adammini'):
117117
pytest.skip(f'skip {optimizer_name}')
118118

119119
param = torch.randn(1, 1).bfloat16().requires_grad_(True)

0 commit comments

Comments
 (0)