(reduce_dims, dimensions)
| 77 | |
| 78 | |
| 79 | def check_reduce_dims(reduce_dims, dimensions): |
| 80 | if reduce_dims is not ...: |
| 81 | if is_scalar(reduce_dims): |
| 82 | reduce_dims = [reduce_dims] |
| 83 | if any(dim not in dimensions for dim in reduce_dims): |
| 84 | raise ValueError( |
| 85 | f"cannot reduce over dimensions {reduce_dims!r}. expected either '...' " |
| 86 | f"to reduce over all dimensions or one or more of {dimensions!r}. " |
| 87 | f"Alternatively, install the `flox` package. " |
| 88 | ) |
| 89 | |
| 90 | |
| 91 | def _codes_to_group_indices(codes: np.ndarray, N: int) -> GroupIndices: |