Correct serie's dtype

theislab · Nov 2, 2023 · 81ac67b · 81ac67b
1 parent fcc8114
commit 81ac67b
Show file tree

Hide file tree

Showing 2 changed files with 2 additions and 2 deletions.
diff --git a/src/cellrank/_utils/_utils.py b/src/cellrank/_utils/_utils.py
@@ -886,7 +886,7 @@ def _convert_to_categorical_series(
             "that there are no conflicting keys, such as `0` and `'0'`."
         )
 
-    term_states = pd.Series([np.nan] * len(cell_names), index=cell_names)
+    term_states = pd.Series([None] * len(cell_names), index=cell_names, dtype=str)
     for ts, cells in mapper.items():
         term_states[cells] = ts
 

diff --git a/tests/test_cflare.py b/tests/test_cflare.py
@@ -415,7 +415,7 @@ def test_compare_fate_probabilities_with_reference(self):
 
         c = cr.estimators.CFLARE(cr.kernels.PrecomputedKernel(transition_matrix))
 
-        state_annotation = pd.Series(index=range(len(c)))
+        state_annotation = pd.Series(index=range(len(c)), dtype=str)
         state_annotation[7] = "terminal_1"
         state_annotation[10] = "terminal_2"
         state_annotation = state_annotation.astype("category")