def add_data_helper()

in viz/utils.py [0:0]

• 53 lines of code
• 14 McCabe index (conditional complexity)

def add_data_helper(dh, ax, data, x, y, label, **kwargs):
    if "ls" in kwargs:
        ls = kwargs["ls"]
        del kwargs["ls"]
    if "marker" in kwargs:
        marker = kwargs["marker"]
        del kwargs["marker"]
    if "color" in kwargs:
        color = kwargs["color"]
        del kwargs["color"]
    if "outlier" in kwargs:
        outlier = kwargs["outlier"]
        del kwargs["outlier"]
    else:
        outlier = lambda x, y, d: False
    if "reverse" in kwargs:
        reverse = kwargs["reverse"]
        del kwargs["reverse"]
    else:
        reverse = False
    if "mirror" in kwargs:
        mirror = kwargs["mirror"]
        del kwargs["mirror"]
    else:
        mirror = False
    if "preprend" in kwargs:
        preprend = kwargs["preprend"]
        del kwargs["preprend"]
    else:
        preprend = None
    curve = query(data, x=x, y=y, outlier=outlier, **kwargs,)
    if preprend is not None:
        curve[1] = preprend[1]
    if reverse:
        if "acc" in y:
            new_curve = query(data, x=x, y="m0_acc", outlier=outlier, **kwargs)
        else:
            new_curve = query(data, x=x, y="ece_m0", outlier=outlier, **kwargs)
        curve.update({1 - k: v for k, v in new_curve.items()})
    if mirror:
        curve.update({1 - k: v for k, v in curve.items()})

    dh.add(
        *add_point(
            ax,
            curve,
            label=label,
            color=color,
            ls=ls,
            linewidth=3,
            marker=marker,
        )
    )

    return curve