version check for 0.1.1 (#141)

* version check * Fix Co-authored-by: Ubuntu <ubuntu@ip-172-31-68-185.ec2.internal> Co-authored-by: Ubuntu <ubuntu@ip-172-31-63-129.ec2.internal>
awslabs · Aug 24, 2020 · b76eb69 · b76eb69
1 parent 32adef9
commit b76eb69
Showing 1 changed file with 5 additions and 0 deletions.
diff --git a/python/dglke/__init__.py b/python/dglke/__init__.py
@@ -20,5 +20,10 @@
 #!/usr/bin/env python3
 
 import pkg_resources
+from distutils.version import LooseVersion
+import dgl
+
+if LooseVersion(dgl.__version__) != '0.4.3':
+    raise Exception("DGL-KE 0.1.1 only work with DGL 0.4.3")
 
 __version__ = pkg_resources.resource_string("dglke", "VERSION.txt").decode("utf-8").strip()