Fix pickling with python3

2ebe821e · Tiago Peixoto · 5d76bbb1 · 2ebe821e
Commit 2ebe821e authored Dec 26, 2012 by Tiago Peixoto
Hide whitespace changes
Inline Side-by-side

Showing with 2 additions and 2 deletions

src/graph_tool/io.py src/graph_tool/io.py +2 -2

No files found.
--- a/src/graph_tool/io.py
+++ b/src/graph_tool/io.py
@@ -63,11 +63,11 @@ libgraph_tool_core.OStream.write = OStream_write
 def pickler(stream, obj):
    sstream = BytesIO()
    pickle.dump(obj, sstream, -1)
-    stream.write(base64.b64encode(sstream.getvalue()))
+    stream.write(base64.b64encode(sstream.getvalue().encode('ascii')))


 def unpickler(stream):
-    sstream = BytesIO(base64.b64decode(stream.read()))
+    sstream = BytesIO(base64.b64decode(stream.read().encode('ascii')))
    return pickle.load(sstream)

 libgraph_tool_core.set_pickler(pickler)