Ensure we return a bytearray in singleton case

dask · Jun 22, 2020 · b73dcd6 · b73dcd6
1 parent d7f58cb
commit b73dcd6
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/distributed/protocol/utils.py b/distributed/protocol/utils.py
@@ -1,7 +1,7 @@
 import struct
 import msgpack
 
-from ..utils import nbytes
+from ..utils import ensure_bytearray, nbytes
 
 BIG_BYTES_SHARD_SIZE = 2 ** 26
 
@@ -82,7 +82,7 @@ def merge_frames(header, frames):
                 frames.append(mv[l:])
                 l = 0
         if len(L) == 1:  # no work necessary
-            out.append(L[0])
+            out.append(ensure_bytearray(L[0]))
         else:
             out.append(bytearray(b"").join(map(memoryview, L)))
     return out