minor bug

IntelCompH2020 · Sep 12, 2023 · 1e2c1e2 · 1e2c1e2
1 parent f52cfdf
commit 1e2c1e2
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/src/task_manager.py b/src/task_manager.py
@@ -1429,7 +1429,7 @@ def export_subcorpus(self):
                             "You must load or create a set of labels first")
             return
 
-        subcorpus = self.df_dataset[self.df_dataset.prediction == 1]
+        subcorpus = self.df_dataset[self.df_dataset.PU_prediction == 1]
 
         # Save ids only
         path2parquet = (