Function _get_single_namespace

xarray/compat/array_api_compat.py:50–60

(x)

def get_array_namespace(*values):
    def _get_single_namespace(x):
        if hasattr(x, "__array_namespace__"):
            return x.__array_namespace__()
        elif isinstance(x, array_type("cupy")):
            # cupy is fully compliant from xarray's perspective, but will not expose
            # __array_namespace__ until at least v14. Special case it for now
            import cupy as cp

            return cp
        else:
            return np

    namespaces = {_get_single_namespace(t) for t in values}
    non_numpy = namespaces - {np}