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 dd81eb5 commit 45fe685Copy full SHA for 45fe685
tests/test_utils.py
@@ -12,6 +12,7 @@
12
normalize_gradient,
13
unit_norm,
14
)
15
+from pytorch_optimizer.version import __VERSION__
16
from tests.utils import Example
17
18
@@ -76,3 +77,7 @@ def test_is_valid_parameters():
76
77
after_parameters = get_optimizer_parameters(model, weight_decay=1e-3, wd_ban_list=wd_ban_list)
78
79
assert is_valid_parameters(after_parameters)
80
+
81
82
+def test_version():
83
+ assert len(__VERSION__.split('.')) == 3
0 commit comments