| 564 | |
| 565 | |
| 566 | class DataNode(GraphNode): |
| 567 | value: Any |
| 568 | typ: type |
| 569 | __slots__ = tuple(__annotations__) |
| 570 | |
| 571 | def __init__(self, key: Any, value: Any): |
| 572 | if key is None: |
| 573 | key = (type(value).__name__, next(_anom_count)) |
| 574 | self.key = key |
| 575 | self.value = value |
| 576 | self.typ = type(value) |
| 577 | self._dependencies = _no_deps |
| 578 | |
| 579 | @property |
| 580 | def data_producer(self) -> bool: |
| 581 | return True |
| 582 | |
| 583 | def copy(self): |
| 584 | return DataNode(self.key, self.value) |
| 585 | |
| 586 | def __call__(self, values=()): |
| 587 | return self.value |
| 588 | |
| 589 | def __repr__(self): |
| 590 | return f"DataNode({self.value!r})" |
| 591 | |
| 592 | def __reduce__(self): |
| 593 | return (DataNode, (self.key, self.value)) |
| 594 | |
| 595 | def __dask_tokenize__(self): |
| 596 | from dask.base import tokenize |
| 597 | |
| 598 | return (type(self).__name__, tokenize(self.value)) |
| 599 | |
| 600 | def substitute( |
| 601 | self, subs: dict[KeyType, KeyType | GraphNode], key: KeyType | None = None |
| 602 | ) -> DataNode: |
| 603 | if key is not None and key != self.key: |
| 604 | return DataNode(key, self.value) |
| 605 | return self |
| 606 | |
| 607 | def __iter__(self): |
| 608 | return iter(self.value) |
| 609 | |
| 610 | |
| 611 | def _get_dependencies(obj: object) -> set | frozenset: |
no outgoing calls