Skip to content

Commit 7de15fc

Browse files
committed
fix: test_betas
1 parent 1446f6c commit 7de15fc

File tree

1 file changed

+9
-4
lines changed

1 file changed

+9
-4
lines changed

tests/test_optimizer_parameters.py

Lines changed: 9 additions & 4 deletions
Original file line numberDiff line numberDiff line change
@@ -120,19 +120,24 @@ def test_betas(optimizer_name):
120120
optimizer = load_optimizer(optimizer_name)
121121

122122
with pytest.raises(ValueError):
123-
optimizer(None, betas=(-0.1, 0.1))
123+
if optimizer_name == 'ranger21':
124+
optimizer(None, num_iterations=100, betas=(-0.1, 0.1))
125+
else:
126+
optimizer(None, betas=(-0.1, 0.1))
124127

125128
with pytest.raises(ValueError):
126-
optimizer(None, betas=(0.1, -0.1))
129+
if optimizer_name == 'ranger21':
130+
optimizer(None, num_iterations=100, betas=(0.1, -0.1))
131+
else:
132+
optimizer(None, betas=(0.1, -0.1))
127133

128134
if optimizer_name == 'adapnm':
129135
with pytest.raises(ValueError):
130136
optimizer(None, betas=(0.1, 0.1, -0.1))
131137

132138

133139
def test_reduction():
134-
model: nn.Module = Example()
135-
parameters = model.parameters()
140+
parameters = Example().parameters()
136141
optimizer = load_optimizer('adamp')(parameters)
137142

138143
with pytest.raises(ValueError):

0 commit comments

Comments
 (0)