(self)
| 625 | return self._cached_dict["dsk"] |
| 626 | |
| 627 | def get_output_keys(self): |
| 628 | if self.output_blocks: |
| 629 | # Culling has already generated a list of output blocks |
| 630 | return {(self.output, *p) for p in self.output_blocks} |
| 631 | |
| 632 | # Return all possible output keys (no culling) |
| 633 | return { |
| 634 | (self.output, *p) |
| 635 | for p in itertools.product( |
| 636 | *[range(self.dims[i]) for i in self.output_indices] |
| 637 | ) |
| 638 | } |
| 639 | |
| 640 | def __getitem__(self, key): |
| 641 | return self._dict[key] |