test(featurizer): fix patch_ver issue

tigergraph · Nov 10, 2023 · d139b8d · d139b8d
1 parent e2d6088
commit d139b8d
Showing 1 changed file with 5 additions and 1 deletion.
diff --git a/tests/test_gds_featurizer.py b/tests/test_gds_featurizer.py
@@ -23,7 +23,11 @@ def test_get_db_version(self):
         major_ver, minor_ver, patch_ver = self.featurizer._get_db_version()
         self.assertIsNotNone(int(major_ver))
         self.assertIsNotNone(int(minor_ver))
-        self.assertIsNotNone(int(patch_ver))
+        try: 
+            patch_ver = int(patch_ver)
+        except:
+            pass
+        self.assertIsNotNone(patch_ver)
         self.assertIsInstance(self.featurizer.algo_ver, str)
 
     def test_get_algo_dict(self):