o i\@sddlZddlZddlZddlZddlmZddlZddlmZddl Z ddl Z e Z e ejddZe ejddZddZd d Zd d Zd dZdS)N) defaultdict)partialEINX_WARN_ON_RETRACEEINX_CACHE_SIZEcCst|tjr t|St|ttBrtdd|DSt|tr-tdd| DSt|t j r9tt |St|t jrKt|j|j|j|jfS|S)Ncss|]}t|VqdS)N _freeze_value).0xr L/home/ubuntu/.local/lib/python3.10/site-packages/einx/_src/util/lru_cache.py sz _freeze_value..cSi|] \}}|t|qSr rr kvr r r z!_freeze_value..) isinstancenpndarrayrtolistlisttupledict frozendictitemstypesSimpleNamespacevarsinspect Parameternamedefault annotationkind)r r r r rs      rcstfdd}|S)Ncs.dd|D}dd|D}|i|S)NcSsg|]}t|qSr r)r ar r r #sz5_freeze_args..func_frozen..cSrr rrr r r r$rz5_freeze_args..func_frozen..)r)argskwargsfuncr r func_frozen!sz!_freeze_args..func_frozen) functoolswraps)r+r,r r*r _freeze_args sr/cs4tdkrtddtfdd}|SS)NrcSsdS)Nrr r r r r ,sz'_with_retrace_warning..c s*d}tdkr{ttdsdt_t}d}t|D] \}}|jtkr#|}q||dd}t |dkr{d}t |D]%}|d|jd|j d |j d 7}|j dur]|d |j dd 7}q8|d7<tjr{|tkr{d}td td ||rdt_|i|}dt_|S|i|}|S)NFrwarnTzFile "z", line z, in  z z9WARNING (einx): The following call stack has resulted in z retraces of an einx function. A retrace happens when the function is called with different signatures of input arguments. Call stack (most recent call last): )warn_on_retrace_numhasattr _thread_localr1r stack enumeratefilename__file__lenreversedlinenofunction code_contextstripprint) r(r) has_warnedr8 last_indexiframetraceresultcache_failuresr+r r func_with_warn.sF    "  z-_with_retrace_warning..func_with_warn)r5rr-r.)r+rKr rIr _with_retrace_warning*s  +rLcCsjt|}tdkrtjtdkrtndd|}ntdkr/dttvr't|}ntjdd|}t|}|S)Nr)maxsizecache)rLmax_cache_sizer- lru_cacherrNr/r*r r r rPbs  rP)osr- threadingr collectionsrnumpyrrrrlocalr7intenvirongetr5rOrr/rLrPr r r r s    8