We read every piece of feedback, and take your input very seriously.
To see all available qualifiers, see our documentation.
There was an error while loading. Please reload this page.
1 parent a57113e commit 030c65aCopy full SHA for 030c65a
tests/test_load_optimizers.py
@@ -9,7 +9,7 @@ def ids(v) -> str:
9
return f'{v[0].__name__}_{v[1:]}'
10
11
12
-OPTIMIZER_NAMES: List[str] = [
+VALID_OPTIMIZER_NAMES: List[str] = [
13
'adamp',
14
'sgdp',
15
'madgrad',
@@ -25,7 +25,6 @@ def ids(v) -> str:
25
]
26
27
INVALID_OPTIMIZER_NAMES: List[str] = [
28
- 'invalid',
29
'asam',
30
'sam',
31
'pcgrad',
@@ -35,7 +34,7 @@ def ids(v) -> str:
35
34
36
37
38
-@pytest.mark.parametrize('valid_optimizer_names', OPTIMIZER_NAMES)
+@pytest.mark.parametrize('valid_optimizer_names', VALID_OPTIMIZER_NAMES)
39
def test_load_optimizers_valid(valid_optimizer_names):
40
load_optimizers(valid_optimizer_names)
41
0 commit comments