Commit 2bb4c286 authored by Jan Kiene's avatar Jan Kiene
Browse files

add argument for selecting the columns to compare on

parent e9d7afef

scripts/basop_check_for_regressions.py

+8 −1
Original line number Diff line number Diff line
@@ -25,11 +25,12 @@ def main(args): 


    pd.options.display.max_colwidth = 200

    regressions_found = False

    for col, thresh in COLS_2_THRESHOLDS.items():

    for col in args.columns_to_compare:

        col_curr = f"{col}-curr"

        col_prev = f"{col}-prev"

        diff = df_merged[col_curr] - df_merged[col_prev]



        thresh = COLS_2_THRESHOLDS[col]

        # invert sign of difference for "higher is better" metrics

        if thresh < 0:

            diff *= -1
@@ -61,6 +62,12 @@ if __name__ == "__main__": 
    parser = argparse.ArgumentParser()

    parser.add_argument("csv_current")

    parser.add_argument("csv_previous")

    parser.add_argument(

        "--columns_to_compare",

        nargs="+",

        choices=COLS_2_THRESHOLDS.keys(),

        default=COLS_2_THRESHOLDS.keys(),

    )



    args = parser.parse_args()

    sys.exit(main(args))