From b3e3cfbd3377644954f808925a23dcb1087a81b6 Mon Sep 17 00:00:00 2001 From: Kentaro Wada Date: Mon, 5 Nov 2018 14:25:49 +0000 Subject: [PATCH] Use importlib instead of imp for Python3 --- setup.py | 13 ++++++++++--- 1 file changed, 10 insertions(+), 3 deletions(-) diff --git a/setup.py b/setup.py index f89d0f7..dca8215 100644 --- a/setup.py +++ b/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 -- GitLab