o i@sUddlZddlZddlmZddlmZmZddlmZeZee e fe d<edZ edZ de d dfd d Zd ee e fd ee e ffd dZdS)N) OrderedDict)CallableTypeVar) ParamSpecsimple_call_counter_P_RlabelreturncCst|d}|dt|<dSNr)r setdefault)r prevrO/home/ubuntu/veenaModal/venv/lib/python3.10/site-packages/torch/utils/_stats.py count_labels rfncs,tdtjdtjdtffdd }|S)Nargskwargsr cs6jtvr dtj<tjdtj<|i|Sr ) __qualname__r)rrrrrwrappers  zcount..wrapper) functoolswrapsrrrr)rrrrrcounts r) collectionsrrtypingrrtyping_extensionsrrstrint__annotations__rrrrrrrrs   &