relocate removed classes file to extraction directory

MannLabs · Jan 11, 2025 · fbac9fe · fbac9fe
1 parent d4663d8
commit fbac9fe
Showing 1 changed file with 2 additions and 1 deletion.
diff --git a/src/scportrait/pipeline/extraction.py b/src/scportrait/pipeline/extraction.py
@@ -450,9 +450,10 @@ def _get_label_info(self, arg):
 
     def _save_removed_classes(self, classes):
         # define path where classes should be saved
+
         filtered_path = os.path.join(
             self.project_location,
-            self.DEFAULT_SEGMENTATION_DIR_NAME,
+            self.DEFAULT_EXTRACTION_DIR_NAME,
             self.DEFAULT_REMOVED_CLASSES_FILE,
         )