(key, data: duckarray[Any, Any] | None = None)
| 1539 | |
| 1540 | |
| 1541 | def _masked_result_drop_slice(key, data: duckarray[Any, Any] | None = None): |
| 1542 | key = (k for k in key if not isinstance(k, slice)) |
| 1543 | chunks_hint = getattr(data, "chunks", None) |
| 1544 | |
| 1545 | new_keys = [] |
| 1546 | for k in key: |
| 1547 | if isinstance(k, np.ndarray): |
| 1548 | if is_chunked_array(data): # type: ignore[arg-type] |
| 1549 | chunkmanager = get_chunked_array_type(data) |
| 1550 | new_keys.append( |
| 1551 | _chunked_array_with_chunks_hint(k, chunks_hint, chunkmanager) |
| 1552 | ) |
| 1553 | elif isinstance(data, array_type("sparse")): |
| 1554 | import sparse |
| 1555 | |
| 1556 | new_keys.append(sparse.COO.from_numpy(k)) |
| 1557 | else: |
| 1558 | new_keys.append(k) |
| 1559 | else: |
| 1560 | new_keys.append(k) |
| 1561 | |
| 1562 | mask = _logical_any(k == -1 for k in new_keys) |
| 1563 | return mask |
| 1564 | |
| 1565 | |
| 1566 | def create_mask( |
no test coverage detected
searching dependent graphs…