Add PY2 and PY3 to __init__

labelme/__init__.py

@@ -1,6 +1,8 @@
 # flake8: noqa
 
 import logging
+import sys
+
 from qtpy import QT_VERSION
 
 
@@ -9,8 +11,13 @@ __appname__ = 'labelme'
 QT5 = QT_VERSION[0] == '5'
 del QT_VERSION
 
+PY2 = sys.version[0] == '2'
+PY3 = sys.version[0] == '3'
+del sys
+
 logging.basicConfig(level=logging.INFO)
 logger = logging.getLogger(__appname__)
+del logging
 
 
 from labelme._version import __version__

labelme/label_file.py

@@ -4,13 +4,11 @@ import os.path
 import sys
 
 from . import logger
+from . import PY2
 from . import utils
 from ._version import __version__
 
 
-PY2 = sys.version_info[0] == 2
-
-
 class LabelFileError(Exception):
     pass