Browse Source

Use importlib instead of imp for Python3

Kentaro Wada 6 years ago
parent
commit
9d5b2e7ae4
1 changed files with 10 additions and 3 deletions
  1. 10 3
      setup.py

+ 10 - 3
setup.py

@@ -1,4 +1,3 @@
-import imp
 import os.path
 from setuptools import find_packages
 from setuptools import setup
@@ -13,8 +12,16 @@ assert PY3 or PY2
 
 
 here = os.path.abspath(os.path.dirname(__file__))
-version = imp.load_source(
-    '_version', os.path.join(here, 'labelme', '_version.py')).__version__
+version_file = os.path.join(here, 'labelme', '_version.py')
+if PY3:
+    import importlib
+    version = importlib.machinery.SourceFileLoader(
+        '_version', version_file
+    ).load_module().__version__
+else:
+    assert PY2
+    import imp
+    version = imp.load_source('_version', version_file).__version__
 del here