Use nonempty_domain to access dimensions of the cell and gene axes (#…

…23) Co-authored-by: Tony Kuo <[email protected]>
BiocPy · Jun 28, 2024 · 963c923 · 963c923
1 parent cb76ed8
commit 963c923
Show file tree

Hide file tree

Showing 2 changed files with 6 additions and 3 deletions.
diff --git a/src/cellarr/CellArrDataset.py b/src/cellarr/CellArrDataset.py
@@ -442,7 +442,10 @@ def __getitem__(
     ####
     @property
     def shape(self):
-        return (self._cell_metadata_tdb.shape[0], self._gene_annotation_tdb.shape[0])
+        return (
+            self._cell_metadata_tdb.nonempty_domain()[0][1] + 1,
+            self._gene_annotation_tdb.nonempty_domain()[0][1] + 1,
+        )
 
     def __len__(self):
         return self.shape[0]

diff --git a/src/cellarr/dataloader.py b/src/cellarr/dataloader.py
@@ -254,8 +254,8 @@ def __init__(
         )
 
         self.matrix_shape = (
-            self.cell_metadata_tdb.shape[0],
-            self.gene_annotation_tdb.shape[0],
+            self.cell_metadata_tdb.nonempty_domain()[0][1] + 1,
+            self.gene_annotation_tdb.nonempty_domain()[0][1] + 1,
         )
 
         # limit to cells with labels