Skip to content

Commit 1a9294d

Browse files
committed
update: test_ranger21_methods
1 parent 464b2dc commit 1a9294d

File tree

1 file changed

+3
-5
lines changed

1 file changed

+3
-5
lines changed

tests/test_optimizer_parameters.py

Lines changed: 3 additions & 5 deletions
Original file line numberDiff line numberDiff line change
@@ -176,8 +176,6 @@ def test_safe_fp16_methods():
176176

177177

178178
def test_ranger21_methods():
179-
warm_up_iterations: int = Ranger21.build_warm_up_iterations(1000, 0.999)
180-
assert warm_up_iterations == 220
181-
182-
warm_down_iterations = Ranger21.build_warm_down_iterations(1000)
183-
assert warm_down_iterations == 280
179+
assert Ranger21.build_warm_up_iterations(1, 0.999) == 0
180+
assert Ranger21.build_warm_up_iterations(1000, 0.999) == 220
181+
assert Ranger21.build_warm_down_iterations(1000) == 280

0 commit comments

Comments
 (0)