diff options
| -rw-r--r-- | test/remote_test.py | 6 |
1 files changed, 5 insertions, 1 deletions
diff --git a/test/remote_test.py b/test/remote_test.py index 1b42f8a0189..d084133d475 100644 --- a/test/remote_test.py +++ b/test/remote_test.py @@ -9,7 +9,11 @@ from pickle import dumps import six from six import moves import sys -from aenum import IntEnum, IntFlag + +if sys.version_info < (3,): + from aenum import IntEnum, IntFlag +else: + from enum import IntEnum, IntFlag class SerializableClassCopy(object): |