From b4ed90ec9bb58bb7ce9ed23b33fe1e8249173855 Mon Sep 17 00:00:00 2001 From: verena <9377970+vpchung@users.noreply.github.com> Date: Wed, 8 May 2024 15:08:01 -0700 Subject: [PATCH] fix column dtype mixup --- score.py | 8 ++++---- 1 file changed, 4 insertions(+), 4 deletions(-) diff --git a/score.py b/score.py index a0157bb..e5144ae 100644 --- a/score.py +++ b/score.py @@ -60,13 +60,13 @@ def main(): if res.get("validation_status") == "VALIDATED": pred = pd.read_csv( args.predictions_file, - usecols=GOLDSTANDARD_COLS, - dtype=GOLDSTANDARD_COLS + usecols=PREDICTION_COLS, + dtype=PREDICTION_COLS ) gold = pd.read_csv( gold_file, - usecols=PREDICTION_COLS, - dtype=PREDICTION_COLS + usecols=GOLDSTANDARD_COLS, + dtype=GOLDSTANDARD_COLS ) scores = score(gold, "disease", pred, "disease_probability") status = "SCORED"