(self)
| 643 | ) |
| 644 | |
| 645 | def __dask_graph__(self): |
| 646 | graphs = {k: v.__dask_graph__() for k, v in self.variables.items()} |
| 647 | graphs = {k: v for k, v in graphs.items() if v is not None} |
| 648 | if not graphs: |
| 649 | return None |
| 650 | else: |
| 651 | try: |
| 652 | from dask.highlevelgraph import HighLevelGraph |
| 653 | |
| 654 | return HighLevelGraph.merge(*graphs.values()) |
| 655 | except ImportError: |
| 656 | from dask import sharedict |
| 657 | |
| 658 | return sharedict.merge(*graphs.values()) |
| 659 | |
| 660 | def __dask_keys__(self): |
| 661 | import dask |