Skip to content

Commit c548a4d

Browse files
committed
update: test case
1 parent 5737268 commit c548a4d

File tree

1 file changed

+4
-8
lines changed

1 file changed

+4
-8
lines changed

tests/test_general_optimizer_parameters.py

Lines changed: 4 additions & 8 deletions
Original file line numberDiff line numberDiff line change
@@ -123,14 +123,10 @@ def test_momentum(optimizer_name):
123123
def test_beta(optimizer_name):
124124
optimizer = load_optimizer(optimizer_name)
125125

126-
if optimizer_name == 'ranger21':
127-
# test beta0
128-
with pytest.raises(ValueError):
129-
optimizer(None, num_iterations=200, beta0=-0.1)
130-
else:
131-
# test beta
132-
with pytest.raises(ValueError):
133-
optimizer(None, beta=-0.1)
126+
config = {'num_iterations': 200, 'beta0': -0.1} if optimizer_name == 'ranger21' else {'beta': -0.1}
127+
128+
with pytest.raises(ValueError):
129+
optimizer(None, **config)
134130

135131

136132
@pytest.mark.parametrize('optimizer_name', BETA_OPTIMIZER_NAMES)

0 commit comments

Comments
 (0)