wip

dask · Mar 20, 2020 · ad5a99b · ad5a99b
1 parent d69511e
commit ad5a99b
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/distributed/comm/ucx.py b/distributed/comm/ucx.py
@@ -183,7 +183,7 @@ async def write(
 
                 for frame in send_frames:
                     await self.ep.send(frame)
-                return sum(map(nbytes, frames))
+                return sum(map(nbytes, send_frames))
             except (ucp.exceptions.UCXBaseException):
                 self.abort()
                 raise CommClosedError("While writing, the connection was closed")