PiddlZddlmZmZmZmZmZejeZ eddZ Gddee Z dS)N)AnyDictIteratorOptionalTypeVarTT) covariantceZdZdZdZdZd deefdZde e e ffdZ de fd Zd Zde e e ffd ZdS) BaseNodeaBaseNodes are the base class for creating composable dataloading DAGs in ``torchdata.nodes``. Most end-users will not iterate over a BaseNode instance directly, but instead wrap it in a :class:`torchdata.nodes.Loader` which converts the DAG into a more familiar Iterable. .. code-block:: python node = MyBaseNodeImpl() loader = Loader(node) # loader supports state_dict() and load_state_dict() for epoch in range(5): for idx, batch in enumerate(loader): ... # or if using node directly: node = MyBaseNodeImpl() for epoch in range(5): node.reset() for idx, batch in enumerate(loader): ... cd|_dS)zPSubclasses must implement this method and call super().__init__(*args, **kwargs)FN_BaseNode__initialized)selfargskwargss m/home/jaya/work/projects/VOICE-AGENT/VIET/agent-env/lib/python3.11/site-packages/torchdata/nodes/base_node.py__init__zBaseNode.__init__(s"c|SNrs r__iter__zBaseNode.__iter__,s rN initial_statecd|_dS)aResets the iterator to the beginning, or to the state passed in by initial_state. Reset is a good place to put expensive initialization, as it will be lazily called when ``next()`` or ``state_dict()`` is called. Subclasses must call ``super().reset(initial_state)``. Args: initial_state: Optional[dict] - a state dict to pass to the node. If None, reset to the beginning. TNr )rrs rresetzBaseNode.reset/s"rreturnc:tt|)zSubclasses must implement this method, instead of ``state_dict()``. Should only be called by BaseNode. Returns: Dict[str, Any] - a state dict that may be passed to ``reset()`` at some point in the future NotImplementedErrortypers r get_statezBaseNode.get_state; "$t**---rc:tt|)zSubclasses must implement this method, instead of ``__next__``. Should only be called by BaseNode. Returns: T - the next value in the sequence, or throw StopIteration rrs rnextz BaseNode.nextCr#rc |jn-#t$r tdt|wxYw|js;|d|jstdt||S)NJself.__initialized not found, did you call super().__init__()? type(self)=fFailed to initialize after .reset(), did you call super().reset() in your .reset() method? type(self)=)rAttributeErrorr r!rr%rs r__next__zBaseNode.__next__Ks w     w w w%&uhlmqhrhr&u&uvv v w!  JJt   % )rvw{r|r|yy{{ *4c |jn-#t$r tdt|wxYw|js;|d|jstdt||S)zGet a state_dict for this BaseNode. Returns: Dict[str, Any] - a state dict that may be passed to ``reset()`` at some point in the future. r'Nr()rr)r r!rr"rs r state_dictzBaseNode.state_dictXs  w     w w w%&uhlmqhrhr&u&uvv v w!  JJt   % )rvw{r|r|~~r+r)__name__ __module__ __qualname____doc__rrrdictrrstrrr"rr%r*r-rrrr r s.### " "8D> " " " ".4S>.....a....    DcN      rr ) loggingtypingrrrrr getLoggerr.loggerrr rrrr8s99999999999999  8 $ $ GC4   Y Y Y Y Y x{Y Y Y Y Y r