(self)
| 76 | |
| 77 | @functools.cached_property |
| 78 | def _cached_keys(self): |
| 79 | out = self.lower_completely() |
| 80 | |
| 81 | name, chunks, numblocks = out.name, out.chunks, out.numblocks |
| 82 | |
| 83 | def keys(*args): |
| 84 | if not chunks: |
| 85 | return List(TaskRef((name,))) |
| 86 | ind = len(args) |
| 87 | if ind + 1 == len(numblocks): |
| 88 | result = List( |
| 89 | *(TaskRef((name,) + args + (i,)) for i in range(numblocks[ind])) |
| 90 | ) |
| 91 | else: |
| 92 | result = List(*(keys(*(args + (i,))) for i in range(numblocks[ind]))) |
| 93 | return result |
| 94 | |
| 95 | return keys() |
| 96 | |
| 97 | def __dask_keys__(self): |
| 98 | key_refs = self._cached_keys |
nothing calls this directly
no test coverage detected