Skip to content

Commit 030c65a

Browse files
committed
update: test_load_optimizers
1 parent a57113e commit 030c65a

File tree

1 file changed

+2
-3
lines changed

1 file changed

+2
-3
lines changed

tests/test_load_optimizers.py

Lines changed: 2 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -9,7 +9,7 @@ def ids(v) -> str:
99
return f'{v[0].__name__}_{v[1:]}'
1010

1111

12-
OPTIMIZER_NAMES: List[str] = [
12+
VALID_OPTIMIZER_NAMES: List[str] = [
1313
'adamp',
1414
'sgdp',
1515
'madgrad',
@@ -25,7 +25,6 @@ def ids(v) -> str:
2525
]
2626

2727
INVALID_OPTIMIZER_NAMES: List[str] = [
28-
'invalid',
2928
'asam',
3029
'sam',
3130
'pcgrad',
@@ -35,7 +34,7 @@ def ids(v) -> str:
3534
]
3635

3736

38-
@pytest.mark.parametrize('valid_optimizer_names', OPTIMIZER_NAMES)
37+
@pytest.mark.parametrize('valid_optimizer_names', VALID_OPTIMIZER_NAMES)
3938
def test_load_optimizers_valid(valid_optimizer_names):
4039
load_optimizers(valid_optimizer_names)
4140

0 commit comments

Comments
 (0)