Skip to content

Commit

Permalink
typing
Browse files Browse the repository at this point in the history
  • Loading branch information
Christian-B committed Dec 12, 2024
1 parent be370bf commit 005a9da
Showing 1 changed file with 2 additions and 2 deletions.
Original file line number Diff line number Diff line change
Expand Up @@ -255,12 +255,12 @@ def _max_send_buffer_keys_per_timestep(
counts = numpy.bincount(
numpy.concatenate(send_buffer_times).astype("int"))
if len(counts):
return max(counts)
return int(numpy.max(counts))
return 0
if len(send_buffer_times):
counts = numpy.bincount(send_buffer_times)
if len(counts):
return n_keys * max(counts)
return n_keys * numpy.max(counts)
return 0
return 0

Expand Down

0 comments on commit 005a9da

Please sign in to comment.