Skip to content

Commit 45fe685

Browse files
committed
update: test_version
1 parent dd81eb5 commit 45fe685

File tree

1 file changed

+5
-0
lines changed

1 file changed

+5
-0
lines changed

tests/test_utils.py

Lines changed: 5 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -12,6 +12,7 @@
1212
normalize_gradient,
1313
unit_norm,
1414
)
15+
from pytorch_optimizer.version import __VERSION__
1516
from tests.utils import Example
1617

1718

@@ -76,3 +77,7 @@ def test_is_valid_parameters():
7677
after_parameters = get_optimizer_parameters(model, weight_decay=1e-3, wd_ban_list=wd_ban_list)
7778

7879
assert is_valid_parameters(after_parameters)
80+
81+
82+
def test_version():
83+
assert len(__VERSION__.split('.')) == 3

0 commit comments

Comments
 (0)