Fixed problem with reduction.loads(BytesIO())

celery · Sep 8, 2016 · 78a5b45 · 78a5b45
1 parent fa56afa
commit 78a5b45
Showing 1 changed file with 2 additions and 0 deletions.
diff --git a/billiard/reduction.py b/billiard/reduction.py
@@ -88,6 +88,8 @@ def loadbuf(cls, buf, protocol=None):
 
         @classmethod
         def loads(self, buf, loads=pickle.loads):
+            if isinstance(buf, io.BytesIO):
+                buf = buf.getvalue()
             return loads(buf)
 register = ForkingPickler.register