Skip to content

Commit

Permalink
Annotate nbytes in update_data as a dict
Browse files Browse the repository at this point in the history
The one case where this variable is used it is a `dict`. So go ahead and
annotate it that way. Should speed up usage of this variable when
cythonized.
  • Loading branch information
jakirkham committed Feb 19, 2021
1 parent 383ea03 commit 0582557
Showing 1 changed file with 6 additions and 1 deletion.
7 changes: 6 additions & 1 deletion distributed/scheduler.py
Original file line number Diff line number Diff line change
Expand Up @@ -5566,7 +5566,12 @@ def add_keys(self, comm=None, worker=None, keys=()):
return "OK"

def update_data(
self, comm=None, who_has=None, nbytes=None, client=None, serializers=None
self,
comm=None,
who_has=None,
nbytes: dict = None,
client=None,
serializers=None,
):
"""
Learn that new data has entered the network from an external source
Expand Down

0 comments on commit 0582557

Please sign in to comment.