Function register_xarray

dask/sizeof.py:260–301

()

@sizeof.register_lazy("xarray")
def register_xarray():
    import sys

    import xarray as xr

    XARRAY_VERSION = Version(xr.__version__)
    XARRAY_GE_2024_02 = XARRAY_VERSION >= Version("2024.02.0")

    @sizeof.register(xr.core.utils.Frozen)
    def xarray_sizeof_frozen(obj):
        return sys.getsizeof(obj) + sizeof(obj.mapping)

    @sizeof.register(xr.DataArray)
    @sizeof.register(xr.Variable)
    def xarray_sizeof_da(obj):
        return sys.getsizeof(obj) + sizeof(obj.data)

    @sizeof.register(xr.Dataset)
    def xarray_sizeof_ds(obj):
        return sys.getsizeof(obj) + sizeof(obj.variables)

    if XARRAY_GE_2024_02:
        xarray_sizeof_da = sizeof.register(xr.NamedArray)(xarray_sizeof_da)

    @sizeof.register(xr.core.indexes.Indexes)
    def xarray_sizeof_indexes(obj):
        return (
            sys.getsizeof(obj)
            + sizeof(obj._index_type)
            + sizeof(obj._indexes)
            + sizeof(obj._variables)
            + sizeof(obj._dims)
        )

    @sizeof.register(xr.core.indexes.PandasIndex)
    def xarray_sizeof_pd_index(obj):
        return (
            sys.getsizeof(obj)
            + sizeof(obj.index)
            + sizeof(obj.dim)
            + sizeof(obj.coord_dtype)
        )


def _register_entry_point_plugins():