diff --git a/gas/cli.py b/gas/cli.py index ba995744..e71fd759 100644 --- a/gas/cli.py +++ b/gas/cli.py @@ -645,8 +645,8 @@ def _f(v): brier = r.get("brier") base_sn34 = r.get("base_sn34_score") aug_sn34 = r.get("aug_sn34_score") - aug_mcc = r.get("aug_binary_mcc") - aug_brier = r.get("aug_binary_brier") + aug_mcc = r.get("aug_mcc") + aug_brier = r.get("aug_brier") bar_len = 20 filled = int(sn34 * bar_len)