Fix python2 & 3 compatibility

konlpy · May 18, 2018 · ff0d2d9 · ff0d2d9
1 parent f4ad963
commit ff0d2d9
Show file tree

Hide file tree

Showing 2 changed files with 14 additions and 7 deletions.
diff --git a/konlpy/csvutils.py b/konlpy/csvutils.py
@@ -1,9 +1,15 @@
-#! /usr/bin/python
 # -*- coding: utf-8 -*-
+from __future__ import absolute_import
 
+import sys
 import csv
 import codecs
-import cStringIO
+
+# Python 2 & 3 Compatibility
+if sys.version_info[0] >= 3:
+    from io import StringIO as cStringIO
+else:
+    import cStringIO
 
 
 def _stringify(s, encoding="utf-8"):

diff --git a/konlpy/data.py b/konlpy/data.py
@@ -1,15 +1,16 @@
-#! /usr/bin/python
 # -*- coding: utf-8 -*-
+from __future__ import absolute_import
 
 import os
 import sys
 import textwrap
-try:
-    import cPickle as pickle
-except:
+
+if sys.version_info[0] >= 3:
     import pickle
+else:
+    import cPickle as pickle
 
-from . import utils
+from konlpy import utils
 
 
 #: A dictionary describing the formats that are supported by