Always cast frames to memoryviews

dask · Jul 28, 2021 · 1b974d0 · 1b974d0
1 parent bf83431
commit 1b974d0
Showing 1 changed file with 5 additions and 4 deletions.
diff --git a/distributed/comm/tcp.py b/distributed/comm/tcp.py
@@ -271,10 +271,11 @@ async def write(self, msg, serializers=None, on_error="message"):
                     if stream._write_buffer is None:
                         raise StreamClosedError()
 
-                    if isinstance(each_frame, memoryview):
-                        # Make sure that `len(data) == data.nbytes`
-                        # See <https://github.com/tornadoweb/tornado/pull/2996>
-                        each_frame = each_frame.cast("B")
+                    each_frame = memoryview(each_frame)
+
+                    # Make sure that `len(data) == data.nbytes`
+                    # See <https://github.com/tornadoweb/tornado/pull/2996>
+                    each_frame = each_frame.cast("B")
 
                     stream._write_buffer.append(each_frame)
                     stream._total_write_index += each_frame_nbytes