Skip to content

Commit 6abce12

Browse files
committed
update: test_adalite_reset
1 parent bab6541 commit 6abce12

File tree

1 file changed

+5
-0
lines changed

1 file changed

+5
-0
lines changed

tests/test_optimizers.py

Lines changed: 5 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -464,6 +464,11 @@ def test_prodigy_reset():
464464
assert str(optimizer) == 'Prodigy'
465465

466466

467+
def test_adalite_reset():
468+
optimizer = load_optimizer('adalite')([simple_zero_rank_parameter(True)])
469+
optimizer.reset()
470+
471+
467472
@pytest.mark.parametrize('pre_conditioner_type', [0, 1, 2])
468473
def test_scalable_shampoo_pre_conditioner_with_svd(pre_conditioner_type, environment):
469474
(x_data, y_data), _, loss_fn = environment

0 commit comments

Comments
 (0)