add missing return_key=True

gustaveroussy · Aug 18, 2024 · fd504bf · fd504bf
1 parent d0a541e
commit fd504bf
Show file tree

Hide file tree

Showing 2 changed files with 6 additions and 4 deletions.
diff --git a/sopa/_constants.py b/sopa/_constants.py
@@ -34,9 +34,11 @@ class SopaKeys:
 
 
 class SopaAttrs:
-    CELL_SEGMENTATION = "for_cell_segmentation"
-    TISSUE_SEGMENTATION = "for_tissue_segmentation"
-    BINS_AGGREGATION = "for_bins_aggregation"
+    CELL_SEGMENTATION = "cell_segmentation_image"
+    TISSUE_SEGMENTATION = "tissue_segmentation_image"
+    BINS_AGGREGATION = "bins_aggregation_shapes"
+    BINS_TABLE = "bins_table"
+    TRANSCRIPTS = "transcripts_dataframe"
     GENE_COLUMN = "feature_key"
 
 

diff --git a/sopa/segmentation/aggregate.py b/sopa/segmentation/aggregate.py
@@ -397,7 +397,7 @@ def count_transcripts(
     Returns:
         An `AnnData` object of shape `(n_cells, n_genes)` with the counts per cell
     """
-    points_key, points = get_spatial_element(sdata.points, key=points_key)
+    points_key, points = get_spatial_element(sdata.points, key=points_key, return_key=True)
 
     if geo_df is None:
         geo_df = get_spatial_element(sdata.shapes, key=shapes_key)