o h @sddlZddlZddlZddlmZddlmZmZddlm Z ddl m Z m Z ddgZ ee e fZeeeedffZd d Zd ed ed eedeefddZdedefddZdded eedefddZded ed eedefddZdS)N) Collection)OptionalUnion)dill_available) IterDataPipe MapDataPipetraverse traverse_dps DataPipeGraphcCsdS)NSTUBr r r j/var/www/html/construction_image-detection-poc/venv/lib/python3.10/site-packages/torch/utils/data/graph.py_stub_unpicklersrscan_obj only_datapipecachereturnc s:t}t|}trddlm}||}nd}gdd}fdd}ttf} zQ| D]} | ||r<| |q.z| Wntj t t fy[trX| nYnwW| D]} | d|rm| dq_tr{ddlm} | dS| D]} | d|r| dqtrddlm} | dww) Nr)PicklercSsd}t|tr i}|D]\}}t|tttfr|||<q |St|ttfr=g}|D]}t|tttfr:||q+|St|tttfrG|}|SN) isinstancedictitemsrrrtuplelistappend) ori_statestatekvr r r getstate_hook's$  z0_list_connected_datapipes..getstate_hookcs8|ks t|vr t|t|tdfS)Nr )idNotImplementedErrorraddr)objrcaptured_connectionsrr r reduce_hook7s  z._list_connected_datapipes..reduce_hook)extendF)ioBytesIOpicklerrdillrrset_reduce_ex_hookset_getstate_hookdump PickleErrorAttributeError TypeErrorr') rrrfp dill_Picklerdrr&datapipe_classescls dill_extendr r$r _list_connected_datapipessZ             r9datapipecCst}t|d|dS)a Traverse the DataPipes and their attributes to extract the DataPipe graph. This only looks into the attribute from each DataPipe that is either a DataPipe and a Python collection object such as ``list``, ``tuple``, ``set`` and ``dict``. Args: datapipe: the end DataPipe of the graph Returns: A graph represented as a nested dictionary, where keys are ids of DataPipe instances and values are tuples of DataPipe instance and the sub-graph T)rr)set_traverse_helper)r:rr r r r ZscCs:d}|s|d7}t|t|durd}t}t|||S)a6 Traverse the DataPipes and their attributes to extract the DataPipe graph. [Deprecated] When ``only_dataPipe`` is specified as ``True``, it would only look into the attribute from each DataPipe that is either a DataPipe and a Python collection object such as ``list``, ``tuple``, ``set`` and ``dict``. Note: This function is deprecated. Please use `traverse_dps` instead. Args: datapipe: the end DataPipe of the graph only_datapipe: If ``False`` (default), all attributes of each DataPipe are traversed. This argument is deprecating and will be removed after the next release. Returns: A graph represented as a nested dictionary, where keys are ids of DataPipe instances and values are tuples of DataPipe instance and the sub-graph zV`traverse` function and will be removed after 1.13. Please use `traverse_dps` instead.zM And, the behavior will be changed to the equivalent of `only_datapipe=True`.NF)warningswarn FutureWarningr;r<)r:rmsgrr r r rls  cCst|ttfstdt|dt|}||vriS||t|||}||ifi}|D]}||d t |||q0|S)Nz.Expected `IterDataPipe` or `MapDataPipe`, but z is found) rrr RuntimeErrortyper r"r9copyupdater<)r:rrdp_idrr5itemr r r r<s   r<r)r(r*r=collections.abcrtypingrrtorch.utils._import_utilsr#torch.utils.data.datapipes.datapiperr__all__DataPiperintrr rboolr;rr9r rr<r r r r s>    C"