# /// script
# requires-python = ">=3.11"
# dependencies = [
#   "xarray[complete]@git+https://github.com/pydata/xarray.git@main",
# ]
# ///
#
# This script automatically imports the development branch of xarray to check for issues.
# Please delete this header if you have _not_ tested this script with `uv run`!

import xarray as xr

xr.show_versions()
d = xr.DataArray(
    [[0, 1], [2, 3]],
    coords={
        "lon": (["ny", "nx"], [[30, 40], [40, 50]]),
        "lat": (["ny", "nx"], [[10, 10], [20, 20]]),
    },
    dims=["ny", "nx"],
)

d.groupby(('lon', 'lat')).mean()  # works
d.groupby('lon').map(lambda x: x)  # works
d.groupby(('lon', 'lat')).map(lambda x: x)  # fails
#   File "xarray/core/nputils.py", line 93, in inverse_permutation
#     inverse_permutation[indices] = np.arange(len(indices), dtype=np.intp)
#     ~~~~~~~~~~~~~~~~~~~^^^^^^^^^
# IndexError: arrays used as indices must be of integer (or boolean) type