[FIX] issue with partial extraction

MannLabs · Feb 12, 2025 · c2c45d4 · c2c45d4
1 parent 27bc4f2
commit c2c45d4
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/src/scportrait/pipeline/extraction.py b/src/scportrait/pipeline/extraction.py
@@ -430,7 +430,7 @@ def _get_classes_to_extract(self) -> None:
                 list(range(len(self.classes_loaded))), self.n_cells, replace=False
             )
             chosen_ids.sort()
-            self.classes = self.classes_loaded[chosen_ids]
+            self.classes = chosen_ids
             self.px_centers = self.centers[chosen_ids]
         else:
             self.classes = self.classes_loaded