a ==ic @srddlZddlZddlZGdddeZGdddeZeaddZdd ZGd d d eZ Gd d d e Z dS)Nc@s:eZdZddZeddZddZddZd d d Zd S) _ContextInfocCs||_||_t|_dSN)cls allow_default threadinglocal _local_stack)selfrrr f/home/droni/.local/share/virtualenvs/DPS-5Je3_V2c/lib/python3.9/site-packages/caffe2/python/context.py__init__ sz_ContextInfo.__init__cCst|jdsg|j_|jjS)Nobj)hasattrrr r r r r _stacks z_ContextInfo._stackcCs|j|dSr)rappendr valuer r r entersz_ContextInfo.entercCs2t|jdksJd|j|j|ks.JdS)NrzContext %s is empty.)lenrrpoprr r r exitsz_ContextInfo.exitTcCsBt|jdkr8|sdS|js*Jd|j|||jdS)Nrz*Context %s is required but none is active.)rrrrr)r requiredr r r get_activesz_ContextInfo.get_activeN)T) __name__ __module__ __qualname__r propertyrrrrr r r r r s  rc@seZdZddZddZdS)_ContextRegistrycCs i|_dSr)_ctxsrr r r r 'sz_ContextRegistry.__init__cCsD||jvr:t|ts"Jd|t|t|td|j|<|j|S)Nz'must be a context managed class, got {})r)r issubclassManagedformatrDefaultManagedr rr r r get*s z_ContextRegistry.getN)rrrr r&r r r r r&srcCstSr)_CONTEXT_REGISTRYr r r r _context_registry4sr(cCsddt|jDS)NcSs*g|]"}t|tr|tkr|tkr|qSr )r!r"r$).0rr r r :sz(_get_managed_classes..)inspectgetmro __class__)r r r r _get_managed_classes9s r.c@s6eZdZdZed ddZddZdd Zd d ZdS) r"z Managed makes the inheritted class a context managed class. class Foo(Managed): ... with Foo() as f: assert f == Foo.current() NTcCsBt|}|dur6t||s2Jd|t|f|S|j|dS)Nz)Wrong context type. Expected: %s, got %s.)r)r(r& isinstancetyper)rrrZctx_infor r r currentKs  zManaged.currentcCs$t|D]}t||q|Sr)r.r(r&rr%r r r __enter__Ts zManaged.__enter__cGs$t|D]}t||qdSr)r.r(r&r)r argsrr r r __exit__Ys zManaged.__exit__cstfdd}|S)Ncs6|i|WdS1s(0YdSrr )r3kwargsfuncr r r wrapper^sz!Managed.__call__..wrapper) functoolswraps)r r7r8r r6r __call__]szManaged.__call__)NT) rrr__doc__ classmethodr1r2r4r;r r r r r"As   r"c@seZdZdZdS)r$zy DefaultManaged is similar to Managed but if there is no parent when current() is called it makes a new one. N)rrrr<r r r r r$esr$) r+rr9objectrrr'r(r.r"r$r r r r s $