cellestial 0.48.1__tar.gz → 0.48.2__tar.gz

{cellestial-0.48.1 → cellestial-0.48.2}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: cellestial
-Version: 0.48.1
+Version: 0.48.2
 Summary: 
 License-Expression: Apache-2.0
 License-File: LICENSE

{cellestial-0.48.1 → cellestial-0.48.2}/cellestial/frames/operations.py

@@ -50,15 +50,13 @@ def _pca_variance_frame(
     n_pcs: int | None = None,
     component_column: str = "Principal Component",
     variance_column: str = "Variance Ratio",
+    use_key: str = "pca",
 ) -> DataFrame:
     """Get PCA variance ratio per principal component as a polars DataFrame."""
     if isinstance(data, AnnData):
-        pca_uns = data.uns.get("pca")
+        pca_uns = data.uns.get(use_key)
         if pca_uns is None or "variance_ratio" not in pca_uns:
-            msg = (
-                "PCA variance ratio is not available on this data. "
-                "Run PCA first (e.g., `sc.tl.pca(data)`)."
-            )
+            msg = f"PCA variance ratio not found under `{use_key}`. Run PCA first."
             raise ValueError(msg)
         variance_ratio = np.asarray(pca_uns["variance_ratio"]).ravel()
         n_available = variance_ratio.size

{cellestial-0.48.1 → cellestial-0.48.2}/cellestial/single/heatmap/dotplot.py

@@ -69,6 +69,7 @@ def dotplot(
     dendrogram: bool = False,
     dendrogram_color: str = "black",
     dendrogram_size: float = 0.5,
+    dendrogram_key: str | None = None,
     dendrogram_kwargs: dict | None = None,
     rectangle_size: float = 0.8,
     rectangle_color: str = "#3f3f3f",
@@ -144,6 +145,9 @@ def dotplot(
         Color of the dendrogram segments.
     dendrogram_size : float, default=0.5
         Size (thickness) of the dendrogram segments.
+    dendrogram_key : str | None, default=None
+        Specific key holding the precomputed dendrogram.
+        By default, `dendrogram_{group_by}` is used.
     dendrogram_kwargs : dict | None, default=None
         Additional parameters to pass to the dendrogram geom_segment.
     rectangle_size : float, default=0.8
@@ -339,7 +343,7 @@ def dotplot(
 
     # DETERMINE: y order of groups
     if dendrogram:
-        y_order_groups, paths = _get_dendrogram(data, group_by)
+        y_order_groups, paths = _get_dendrogram(data, group_by, use_key=dendrogram_key)
     else:
         y_order_groups = (
             frame.select(group_by).unique(maintain_order=True)[group_by].cast(pl.String).to_list()

{cellestial-0.48.1 → cellestial-0.48.2}/cellestial/single/heatmap/heatmap.py

@@ -209,6 +209,7 @@ def heatmap(
     group_lines_size: float = 0.6,
     dendrogram_color: str = "black",
     dendrogram_size: float = 0.5,
+    dendrogram_key: str | None = None,
     group_lines_kwargs: dict | None = None,
     dendrogram_kwargs: dict | None = None,
     key_labels: bool = True,
@@ -284,6 +285,9 @@ def heatmap(
         Color of the dendrogram segments.
     dendrogram_size : float, default=0.5
         Size (thickness) of the dendrogram segments.
+    dendrogram_key : str | None, default=None
+        Specific key holding the precomputed dendrogram.
+        By default, `dendrogram_{group_by}` is used.
     group_lines_kwargs : dict | None, default=None
         Additional parameters to pass to the group separator lines geom_segment.
     dendrogram_kwargs : dict | None, default=None
@@ -499,7 +503,7 @@ def heatmap(
 
     # DETERMINE: y order of groups
     if dendrogram:
-        y_order_groups, paths = _get_dendrogram(data, group_by)
+        y_order_groups, paths = _get_dendrogram(data, group_by, use_key=dendrogram_key)
     else:
         y_order_groups = (
             frame.select(group_by).unique(maintain_order=True)[group_by].cast(pl.String).to_list()
@@ -662,6 +666,7 @@ def matrixplot(
     group_lines_size: float = 0.6,
     dendrogram_color: str = "black",
     dendrogram_size: float = 0.5,
+    dendrogram_key: str | None = None,
     group_lines_kwargs: dict | None = None,
     dendrogram_kwargs: dict | None = None,
     key_labels: bool = True,
@@ -731,6 +736,9 @@ def matrixplot(
         Color of the dendrogram segments.
     dendrogram_size : float, default=0.5
         Size (thickness) of the dendrogram segments.
+    dendrogram_key : str | None, default=None
+        Specific key holding the precomputed dendrogram.
+        By default, `dendrogram_{group_by}` is used.
     group_lines_kwargs : dict | None, default=None
         Additional parameters to pass to the group separator lines geom_segment.
     dendrogram_kwargs : dict | None, default=None
@@ -842,6 +850,7 @@ def matrixplot(
         group_lines_size=group_lines_size,
         dendrogram_color=dendrogram_color,
         dendrogram_size=dendrogram_size,
+        dendrogram_key=dendrogram_key,
         group_lines_kwargs=group_lines_kwargs,
         dendrogram_kwargs=dendrogram_kwargs,
         value_column=value_column,

{cellestial-0.48.1 → cellestial-0.48.2}/cellestial/single/heatmap/stacked_violin.py

@@ -199,6 +199,7 @@ def stacked_violin(
     dendrogram: bool = False,
     dendrogram_color: str = "black",
     dendrogram_size: float = 0.5,
+    dendrogram_key: str | None = None,
     dendrogram_kwargs: dict | None = None,
     rectangle: bool = True,
     rectangle_size: float = 0.8,
@@ -292,6 +293,9 @@ def stacked_violin(
         Color of the dendrogram segments.
     dendrogram_size : float, default=0.5
         Size (thickness) of the dendrogram segments.
+    dendrogram_key : str | None, default=None
+        Specific key holding the precomputed dendrogram.
+        By default, `dendrogram_{group_by}` is used.
     dendrogram_kwargs : dict | None, default=None
         Additional parameters to pass to the dendrogram geom_path.
     rectangle : bool, default=True
@@ -456,7 +460,7 @@ def stacked_violin(
 
     # DETERMINE: y order of groups (dendrogram or first-seen order)
     if dendrogram:
-        y_order_groups, paths = _get_dendrogram(data, group_by)
+        y_order_groups, paths = _get_dendrogram(data, group_by, use_key=dendrogram_key)
     else:
         y_order_groups = (
             frame.select(group_by).unique(maintain_order=True)[group_by].cast(pl.String).to_list()

{cellestial-0.48.1 → cellestial-0.48.2}/cellestial/single/quick/elbow.py

@@ -36,6 +36,7 @@ def elbow(
     label: bool = False,
     label_size: float = 4.0,
     label_every: int = 5,
+    pca_key: str = "pca",
     **geom_kwargs,
 ) -> PlotSpec:
     """
@@ -73,6 +74,8 @@ def elbow(
     label_every : int, default=5
         Only components whose number is a multiple of this value are labeled
         (used when `label=True`). For example, `5` labels PC5, PC10, PC15, ...
+    pca_key : str, default='pca'
+        Specific key holding the PCA results.
     **geom_kwargs
         Additional parameters for the `geom_point` layer.
         For more information on geom_point parameters, see:
@@ -124,6 +127,7 @@ def elbow(
         n_pcs=n_pcs,
         component_column=component_column,
         variance_column=variance_column,
+        use_key=pca_key,
     )
 
     # HANDLE: mapping

{cellestial-0.48.1 → cellestial-0.48.2}/cellestial/spatial/spatial.py

@@ -104,7 +104,15 @@ def _spatial_components(
 
         library = spatial_uns[library_id]
         scale_factors = library.get("scalefactors", {})
-        scale_factor = scale_factors.get(f"tissue_{image_key}_scalef", 1.0)
+        scalef_key = f"tissue_{image_key}_scalef"
+        scale_factor = scale_factors.get(scalef_key)
+        if scale_factor is None:
+            if image:
+                _warn(
+                    f"Scale factor `{scalef_key}` missing; "
+                    "spots may not align with the image."
+                )
+            scale_factor = 1.0
         spot_coordinates = data.obsm[spatial_key] * scale_factor
 
         image_array = None

{cellestial-0.48.1 → cellestial-0.48.2}/cellestial/util/dendrogram.py

@@ -4,22 +4,21 @@ import polars as pl
 from anndata import AnnData
 
 
-def _get_dendrogram(data: AnnData, group_by: str) -> tuple[list[str], pl.DataFrame]:
-    """
-    Get or compute the dendrogram for a group_by key and extract paths.
-
-    Checks ``data.uns[f'dendrogram_{group_by}']``.
-    For AnnData, if not present, runs ``scanpy.tl.dendrogram`` to compute it.
-    Extracts the icoord/dcoord arrays into a Polars DataFrame of paths with normalized positions.
-    """
+def _get_dendrogram(
+    data: AnnData,
+    group_by: str,
+    *,
+    use_key: str | None = None,
+) -> tuple[list[str], pl.DataFrame]:
+    """Get or compute the dendrogram for `group_by` and extract paths."""
     import numpy as np
 
     if isinstance(data, AnnData):
-        key = f"dendrogram_{group_by}"
+        key = use_key if use_key is not None else f"dendrogram_{group_by}"
         if key not in data.uns:
             import scanpy as sc
 
-            sc.tl.dendrogram(data, groupby=group_by)
+            sc.tl.dendrogram(data, groupby=group_by, key_added=key)
 
         dendro = data.uns[key]
         categories_ordered = list(dendro["categories_ordered"])

{cellestial-0.48.1 → cellestial-0.48.2}/cellestial/util/utilities.py

@@ -591,12 +591,16 @@ def _resolve_embedding_key(
 
     target = f"X_{dimensions.upper()}"
     needed = max(xy)
+    search_targets = (target, dimensions.upper())  # accept bare `UMAP` too, not only `X_UMAP`
 
     for key in candidates:
-        if key.upper() == target:
-            return target
+        if key.upper() in search_targets:
+            return key.upper()
 
-    prefix_matches = [key for key in candidates if key.upper().startswith(target)]
+    prefix_matches = [
+        key for key in candidates
+        if any(key.upper().startswith(t) for t in search_targets)
+    ]
     if not prefix_matches:
         msg = (
             f"No embedding found for `dimensions='{dimensions}'`\n"

{cellestial-0.48.1 → cellestial-0.48.2}/pyproject.toml

@@ -1,6 +1,6 @@
 [project]
 name = "cellestial"
-version = "0.48.1"
+version = "0.48.2"
 description = ""
 authors = [
     { name = "Zaf4", email = "zafermolbio@gmail.com" },