Function _is_cudf_dataframe

python/ray/data/block.py:134–148

Check if the object is a cudf.DataFrame (lazy import). Checks ``sys.modules`` first to avoid importing cudf (which loads CUDA and ~1.5 GiB of RSS) when it hasn't been imported yet. If cudf is not in ``sys.modules``, no object in the process can be a cudf DataFrame.

(obj: Any)

def _is_cudf_dataframe(obj: Any) -> bool:
    """Check if the object is a cudf.DataFrame (lazy import).

    Checks ``sys.modules`` first to avoid importing cudf (which loads CUDA
    and ~1.5 GiB of RSS) when it hasn't been imported yet.  If cudf is not
    in ``sys.modules``, no object in the process can be a cudf DataFrame.
    """
    if "cudf" not in sys.modules:
        return False
    try:
        import cudf

        return isinstance(obj, cudf.DataFrame)
    except ImportError:
        return False


def _is_empty_schema(schema: Optional[Schema]) -> bool: