U ‰dZã@s:ddlZddlZddlZe ¡aejdd„ƒZdd„ZdS)éNcks~t|tƒs t|ƒstdƒ‚|g}t tƒ¡}|D]<}t|ƒsDtdƒ‚|j}||kr\itj |<tj |  |¡q0dV|t_ dS)NzBarg_scope is only supporting single or a list of helper functions.zAarg_scope is only supporting a list of callable helper functions.) Ú isinstanceÚlistÚcallableÚAssertionErrorÚcopyÚdeepcopyÚget_current_scopeÚ__name__Ú_threadlocal_scopeÚ current_scopeÚupdate)Zsingle_helper_or_listÚkwargsZ old_scopeÚhelperZ helper_key©rúC/tmp/pip-unpacked-wheel-ua33x9lu/caffe2/python/helpers/arg_scope.pyÚ arg_scope s   ÿ  ÿ rcCsttdƒsit_tjS)Nr )Úhasattrr r rrrrrs r)Ú contextlibrÚ threadingÚlocalr ÚcontextmanagerrrrrrrÚs