FazBrowse GitHub Viewer | Trending |
URL:
| Home
Tools: [Download Repo ZIP]   [Original HTTPS Page]

use packaging.version to parse automl package version · HpcDataLab/docker-python@4d2dfd7 · GitHub

Commit 4d2dfd7

Browse files
committed
use packaging.version to parse automl package version
1 parent ab18edf commit 4d2dfd7

1 file changed

Lines changed: 2 additions & 8 deletions

File tree

‎tests/test_automl.py‎

Lines changed: 2 additions & 8 deletions
Original file line numberDiff line numberDiff line change
@@ -5,6 +5,7 @@
55
from kaggle_gcp import KaggleKernelCredentials, init_automl
66
from test.support import EnvironmentVarGuard
77
from google.cloud import storage, automl_v1beta1 as automl
8+
from packaging.version import parse
89

910
def _make_credentials():
1011
import google.auth.credentials
@@ -14,14 +15,7 @@ class TestAutoMl(unittest.TestCase):
1415

1516
def test_version(self):
1617
self.assertIsNotNone(automl.auto_ml_client._GAPIC_LIBRARY_VERSION)
17-
# follows semver (i.e. major.minor.patch)
18-
version_parts = automl.auto_ml_client._GAPIC_LIBRARY_VERSION.split('.')
19-
self.assertEqual(3, len(version_parts))
20-
# Ensures that version is >= 0.5
21-
major = int(version_parts[0])
22-
minor = int(version_parts[1])
23-
if major == 0:
24-
self.assertGreaterEqual(minor, 5)
18+
self.assertGreaterEqual(parse(automl.auto_ml_client._GAPIC_LIBRARY_VERSION), parse("0.5.0"))
2519

2620
class FakeClient:
2721
def __init__(self, credentials=None, client_info=None, **kwargs):

0 commit comments

Comments
 (0)

Back | FazBrowse Home | New Git URL