Function is_data_primitive

marimo/_runtime/primitives.py:59–102

(value: Any)

def is_data_primitive(value: Any) -> bool:
    if is_primitive(value):
        return True

    if inspect.isclass(value):
        return False

    if not (
        hasattr(value, "__array__")
        or hasattr(value, "toarray")
        or hasattr(value, "__array_interface__")
    ):
        return False

    # Handle cross device tensors particular to torch
    # Transfer may be expensive, so non-cpu tensors are considered
    # unhashable.
    if is_instance_by_name(value, "torch.Tensor"):
        return str(value.device) == "cpu"

    # If a numpy like array, ensure that it's not an object array.
    if hasattr(value, "dtype"):
        return not (
            value.dtype is None
            or (hasattr(value.dtype, "hasobject") and value.dtype.hasobject)
        )
    elif hasattr(value, "dtypes"):
        # Bit of discrepancy between objects like polars and pandas, so use
        # narwhals to normalize the dataframe.
        import narwhals.stable.v2 as nw

        try:
            return bool(
                all(
                    dtype.is_numeric()
                    for dtype in nw.from_native(value)
                    .collect_schema()
                    .dtypes()
                )
            )
        except Exception:
            return False
    # Otherwise may be a closely related array object
    return True


def _is_primitive_container(