From 0fd8afa87cdc4e19777488747d6ac730c05b3141 Mon Sep 17 00:00:00 2001 From: Michael Whittaker Date: Thu, 4 Feb 2021 11:41:44 -0800 Subject: [PATCH] Updated viz with flag for limits. --- quorums/viz.py | 6 ++++-- 1 file changed, 4 insertions(+), 2 deletions(-) diff --git a/quorums/viz.py b/quorums/viz.py index 953be0f..b9c4f7f 100644 --- a/quorums/viz.py +++ b/quorums/viz.py @@ -85,11 +85,13 @@ def plot_node_throughput_on(ax: plt.Axes, strategy: Strategy[T], nodes: Optional[List[Node[T]]] = None, read_fraction: Optional[Distribution] = None, - write_fraction: Optional[Distribution] = None): + write_fraction: Optional[Distribution] = None, + draw_node_capacities: bool = True): nodes = nodes or list(strategy.nodes()) d = distribution.canonicalize_rw(read_fraction, write_fraction) fr = sum(weight * fr for (fr, weight) in d.items()) fw = 1 - fr + # TODO(mwhittaker): Explain. node_limits = [ fr * strategy.x_read_probability[node.x] / node_load + fw * strategy.x_write_probability[node.x] / node_load @@ -107,7 +109,7 @@ def plot_node_throughput_on(ax: plt.Axes, scale_by_node_capacity=False, read_fraction=read_fraction, write_fraction=write_fraction, - node_limits=node_limits) + node_limits=node_limits if draw_node_capacities else None) def _plot_node_load_on(ax: plt.Axes,