diff --git a/nnfwtbn/plot.py b/nnfwtbn/plot.py index 40690816efed1a2f628c749ad3b1204e39423f8c..12d6bd223c1f75b4f6aea4df4d515f17960f2c9d 100644 --- a/nnfwtbn/plot.py +++ b/nnfwtbn/plot.py @@ -299,6 +299,7 @@ def uhepp_hist(dataframe, variable, bins, stacks, selection=None, uhepp_obj.ratio.append(uhepp_ritem) + ################## # Compute delayed dask yields, = dask.compute(yields) for key, item in yields.items(): @@ -306,7 +307,14 @@ def uhepp_hist(dataframe, variable, bins, stacks, selection=None, # TODO: Reorder processes if y_log - # TODO: vertical and horizontal lines + ################## + # Vertical lines + for vline in vlines: + if isinstance(vline, (int, float)): + uhepp_obj.v_lines.append(uhepp.VLine(vline)) + else: + pos_x = vline.pop("x") + uhepp_obj.v_lines.append(uhepp.VLine(pos_x, **vline)) if return_uhepp: return uhepp_obj