fix issues with current selection workflow

MannLabs · Dec 12, 2024 · d6c0721 · d6c0721
1 parent 0346cb7
commit d6c0721
Showing 1 changed file with 3 additions and 2 deletions.
diff --git a/src/scportrait/pipeline/selection.py b/src/scportrait/pipeline/selection.py
@@ -155,9 +155,10 @@ class will automatically provide the most recent segmentation together with the
                 segmentation = tempmmap.array(
                     shape=(x, y), dtype=hdf_labels.dtype, tmp_dir_abs_path=self._tmp_dir_path
                 )
-                segmentation = hdf_labels[self.config["segmentation_channel"], :, :]
+                segmentation[:] = hdf_labels[self.config["segmentation_channel"], :, :]
 
-            coord_index = _create_coord_index_sparse(segmentation)
+            coord_index = dict(_create_coord_index_sparse(segmentation))
+
             with open(self.coord_pickle_file_path, "wb") as f:
                 pickle.dump(coord_index, f)
             self.log(f"Coordinate lookup index saved to file {self.coord_pickle_file_path}.")