Merge branch 'main' into update/nb

theislab · Jul 3, 2024 · 7b799cb · 7b799cb
2 parents f2968f3 + 5017e11
commit 7b799cb
Show file tree

Hide file tree

Showing 2 changed files with 1 addition and 0 deletions.
diff --git a/src/moscot/problems/time/_mixins.py b/src/moscot/problems/time/_mixins.py
@@ -1032,6 +1032,7 @@ def temporal_key(self: TemporalMixinProtocol[K, B], key: Optional[str]) -> None:
             return
         if key not in self.adata.obs:
             raise KeyError(f"Unable to find temporal key in `adata.obs[{key!r}]`.")
+        self.adata.obs[key] = self.adata.obs[key].astype("category")
         col = self.adata.obs[key]
         if not (is_categorical_dtype(col) and is_numeric_dtype(col.cat.categories)):
             raise TypeError(

diff --git a/tests/data/moscot_temporal_tests.h5ad b/tests/data/moscot_temporal_tests.h5ad