fix cpickle to cPickle

nladuo · Apr 22, 2017 · 324c3f6 · 324c3f6
1 parent 1eb878a
commit 324c3f6
Show file tree

Hide file tree

Showing 5 changed files with 5 additions and 4 deletions.
diff --git a/weibo.cn/tensorflow-impl/check_dataset.py b/weibo.cn/tensorflow-impl/check_dataset.py
@@ -4,7 +4,7 @@
 from __future__ import print_function
 
 try:
-    import cpickle as pickle
+    import cPickle as pickle
 except ImportError:
     import pickle
 

diff --git a/weibo.cn/tensorflow-impl/common.py b/weibo.cn/tensorflow-impl/common.py
@@ -6,7 +6,7 @@
 """
 import os
 try:
-    import cpickle as pickle
+    import cPickle as pickle
 except ImportError:
     import pickle
 

diff --git a/weibo.cn/tensorflow-impl/format_dataset.py b/weibo.cn/tensorflow-impl/format_dataset.py
@@ -8,7 +8,7 @@
 import sys
 import io
 try:
-    import cpickle as pickle
+    import cPickle as pickle
 except ImportError:
     import pickle
 

diff --git a/weibo.cn/tensorflow-impl/requirements.txt b/weibo.cn/tensorflow-impl/requirements.txt
@@ -1,3 +1,4 @@
+tensorflow == 1.0
 opencv-python
 scikit-learn
 scipy

diff --git a/weibo.cn/tensorflow-impl/train.py b/weibo.cn/tensorflow-impl/train.py
@@ -6,7 +6,7 @@
 
 import os
 try:
-    import cpickle as pickle
+    import cPickle as pickle
 except ImportError:
     import pickle