Spark config

MantisAI · Sep 11, 2023 · 13dbf67 · 13dbf67
1 parent c8584bd
commit 13dbf67
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/grants_tagger_light/retagging/retagging.py b/grants_tagger_light/retagging/retagging.py
@@ -81,7 +81,7 @@ def _create_pipelines(save_to_path, batch_size, train_df, test_df, tag):
         answer = input("Classifier already trained. Do you want to reuse it? [y|n]: ")
         while answer not in ['y', 'n']:
             answer = input("Classifier already trained. Do you want to reuse it? [y|n]: ")
-        if answer == 'n':
+        if answer == 'y':
             retrain = False
 
     if retrain: