fix view_bug

chemprop · Sep 10, 2020 · 298527a · 298527a
1 parent 4c8f5e7
commit 298527a
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/chemprop/web/app/views.py b/chemprop/web/app/views.py
@@ -255,7 +255,7 @@ def train():
 
         # Run training
         logger = create_logger(name=TRAIN_LOGGER_NAME, save_dir=args.save_dir, quiet=args.quiet)
-        task_scores = run_training(args, logger)
+        task_scores = run_training(args, data, logger)[args.metrics[0]]
         process.join()
 
         # Reset globals