fix import error in python3.5

shaoyn0817 · Jun 21, 2018 · 4729cee · 4729cee
1 parent 1529a15
commit 4729cee
Show file tree

Hide file tree

Showing 2 changed files with 4 additions and 4 deletions.
diff --git a/main.py b/main.py
@@ -2,7 +2,7 @@
 # @Author: Jie
 # @Date:   2017-06-15 14:11:08
 # @Last Modified by:   Jie Yang,     Contact: [email protected]
-# @Last Modified time: 2018-06-16 16:51:28
+# @Last Modified time: 2018-06-22 00:01:23
 
 from __future__ import print_function
 import time
@@ -21,7 +21,7 @@
 
 try:
     import cPickle as pickle
-except ModuleNotFoundError:
+except ImportError:
     import pickle
 
 

diff --git a/utils/data.py b/utils/data.py
@@ -2,7 +2,7 @@
 # @Author: Jie
 # @Date:   2017-06-14 17:34:32
 # @Last Modified by:   Jie Yang,     Contact: [email protected]
-# @Last Modified time: 2018-04-26 13:58:10
+# @Last Modified time: 2018-06-22 00:01:47
 from __future__ import print_function
 from __future__ import absolute_import
 import sys
@@ -11,7 +11,7 @@
 
 try:
     import cPickle as pickle
-except ModuleNotFoundError:
+except ImportError:
     import pickle as pickle