bug fix args

4655684b · mibaumgartner · 65594170 · 4655684b
Commit 4655684b authored May 05, 2021 by mibaumgartner
Show whitespace changes
Inline Side-by-side

Showing with 1 addition and 1 deletion

scripts/predict.py scripts/predict.py +1 -1

No files found.
--- a/scripts/predict.py
+++ b/scripts/predict.py
@@ -180,7 +180,7 @@ def main():
    training_dir = get_training_dir(task_model_dir / task_name / model, fold)

    process = args.no_preprocess
-    if test_split:
+    if test_split and process:
        raise ValueError("When using the test split option raw data is not "
                         "supported. Need to add --no_preprocess flag!")