- merged PLD changes, now it works with python 3.x

[tld-builder.git] / TLD_Builder / wrap.py

diff --git a/TLD_Builder/wrap.py b/TLD_Builder/wrap.py
index 0c9df39281cf6d9f724f9d308ba2b573a762e2ff..9f62992dcfad4c3e6806bf8409cd475cda196dae 100644 (file)
--- a/TLD_Builder/wrap.py
+++ b/TLD_Builder/wrap.py
@@ -3,7 +3,11 @@
 import sys
 import log
 import traceback
-import StringIO
+import sys
+if sys.version_info[0] == 2:
+    import StringIO
+else:
+    from io import StringIO
 import os
 import time