Skip to content

Commit

Permalink
fix: black
Browse files Browse the repository at this point in the history
  • Loading branch information
drodarie committed Aug 12, 2024
1 parent 12cb853 commit 25de4aa
Showing 1 changed file with 5 additions and 1 deletion.
6 changes: 5 additions & 1 deletion bsb_nest/connection.py
Original file line number Diff line number Diff line change
Expand Up @@ -97,7 +97,11 @@ def predict_mem_iterator(self, pre_nodes, post_nodes, cs):
predicted_all_mem = (
len(pre_nodes) * 8 * 2 + len(post_nodes) * 8 * 2 + len(cs) * 6 * 8 * (16 + 2)
) * MPI.get_size()
predicted_local_mem = (predicted_all_mem / len(cs.get_local_chunks("out"))) if len(cs.get_local_chunks("out")) > 0 else 0.
predicted_local_mem = (
(predicted_all_mem / len(cs.get_local_chunks("out")))
if len(cs.get_local_chunks("out")) > 0
else 0.0
)
if predicted_local_mem > avmem / 2:
# Iterate block-by-block
return self.block_iterator(cs)
Expand Down

0 comments on commit 25de4aa

Please sign in to comment.