Fixing ignore columns

chemprop · Jul 21, 2020 · 81ecf38 · 81ecf38
1 parent 98864b5
commit 81ecf38
Show file tree

Hide file tree

Showing 2 changed files with 2 additions and 0 deletions.
diff --git a/chemprop/sklearn_predict.py b/chemprop/sklearn_predict.py
@@ -23,6 +23,7 @@ def predict_sklearn(args: SklearnPredictArgs) -> None:
         path=args.test_path,
         smiles_column=args.smiles_column,
         target_columns=[],
+        ignore_columns=[],
         store_row=True
     )
 

diff --git a/chemprop/train/make_predictions.py b/chemprop/train/make_predictions.py
@@ -54,6 +54,7 @@ def make_predictions(args: PredictArgs, smiles: List[str] = None) -> List[List[O
             path=args.test_path,
             args=args,
             target_columns=[],
+            ignore_columns=[],
             skip_invalid_smiles=False,
             store_row=True
         )