Web fix

chemprop · Aug 20, 2020 · 15aaef2 · 15aaef2
1 parent c5d3545
commit 15aaef2
Showing 1 changed file with 4 additions and 1 deletion.
diff --git a/chemprop/web/app/views.py b/chemprop/web/app/views.py
@@ -22,7 +22,7 @@
 
 from chemprop.args import PredictArgs, TrainArgs
 from chemprop.constants import MODEL_FILE_NAME, TRAIN_LOGGER_NAME
-from chemprop.data import get_data, get_header, get_smiles, validate_data
+from chemprop.data import get_data, get_header, get_smiles, get_task_names, validate_data
 from chemprop.train import make_predictions, run_training
 from chemprop.utils import create_logger, load_task_names, load_args
 
@@ -209,6 +209,9 @@ def train():
         '--ensemble_size', str(ensemble_size)
     ])
 
+    # Get task names
+    args.task_names = get_task_names(path=data_path)
+
     # Check if regression/classification selection matches data
     data = get_data(path=data_path)
     targets = data.targets()