Allow choosing nan string (#33)

* allow choosing nan string * rename back to matrix --------- Co-authored-by: Tony Kuo <[email protected]>
BiocPy · Jul 3, 2024 · 18bc076 · 18bc076
1 parent 21419c0
commit 18bc076
Showing 1 changed file with 6 additions and 1 deletion.
diff --git a/src/cellarr/dataloader.py b/src/cellarr/dataloader.py
@@ -184,6 +184,7 @@ def __init__(
         num_workers: int = 0,
         lognorm: bool = True,
         target_sum: float = 1e4,
+        nan_string: str = "nan",
     ):
         """Initialize a ``DataModule``.
 
@@ -228,6 +229,10 @@ def __init__(
 
             target_sum:
                 Target sum for log-normalization.
+
+            nan_string:
+                A string representing NaN.
+                Defaults to "nan".
         """
 
         super().__init__()
@@ -260,7 +265,7 @@ def __init__(
         )
 
         # limit to cells with labels
-        query_condition = f"{self.label_column_name} != 'nan'"
+        query_condition = f"{self.label_column_name} != '{nan_string}'"
         self.data_df = subset_frame(
             self.cell_metadata_tdb,
             query_condition,