try semver

Author: gtfierro

btrdb/utils/ray.py

@@ -2,6 +2,8 @@
 
 import ray
 
+import semver
+
 import btrdb
 from btrdb.conn import BTrDB
 
@@ -24,13 +26,16 @@ def register_serializer(conn_str=None, apikey=None, profile=None):
     """
     assert ray.is_initialized(), "Need to call ray.init() before registering custom serializer"
     # TODO: check the version using the 'semver' package?
-    if ray.__version__ == "0.8.4":
+    ver = semver.VersionInfo.parse(ray.__version__)
+    if ver.major == 0:
         ray.register_custom_serializer(
         BTrDB, serializer=btrdb_serializer, deserializer=partial(btrdb_deserializer, conn_str=conn_str, apikey=apikey, profile=profile))
-    elif ray.__version__ in ["1.2.0", "1.3.0"]:
+    elif ver.major == 1 and ver.minor in range(2, 4):
     # TODO: check different versions of ray?
         ray.util.register_serializer(
         BTrDB, serializer=btrdb_serializer, deserializer=partial(btrdb_deserializer, conn_str=conn_str, apikey=apikey, profile=profile))
+    else:
+        raise Exception("Ray version %s does not have custom serialization. Please upgrade to >= 1.2.0" % ray.__version__)
 
 def btrdb_serializer(_):
     """