L iUdZddlZddlZddlmZddlZGddZeZdZdZ ejjZ iZ eeeeefed<iZeeeefed <d Zd Zd Zd Zej.Zej.ed<ej.Zej.ed<dZdZdZdZy)zJIT-related state. This module stores various pieces of Python-global state relating to the JIT. This is not intended to be imported directly; please the exposed functionalities in `torch.jit`. N)Anyc$eZdZdZddZdZdZy) EnabledProxyzyStores whether the JIT is enabled or not. This is just a wrapper for a bool, so that we get reference semantics Nc6|jdddd|_y)N PYTORCH_JITTz> Using PyTorch JITz> PyTorch JIT DISABLED) parse_envenabledselfs V/mnt/ssd/data/python-lab/Trading/venv/lib/python3.12/site-packages/torch/jit/_state.py__init__zEnabledProxy.__init__s~~ 4!68P  ctjj|}||S|jdvry|jdvry|dk(r t |y|dk(r t |yt d|d) N>1yestrueT>0nofalseF1v0vzUnknown setting of z. Try using 0 or 1.)osenvirongetlowerprint ValueError)r namedefault true_message false_messagevalues r rzEnabledProxy.parse_envs| t$ =N ;;=0 0 [[]2 2 D= ,  d] - .tf4GHIIrc|jSN)r r s r __bool__zEnabledProxy.__bool__,s ||r)returnN)__name__ __module__ __qualname____doc__r rr%rr rrs  J rrcdt_y)NF_enabledr r+rr disabler/3s Hrcdt_y)NTr-r+rr enabler17s Hr_script_classes_name_to_pyclasscD|t|<|t|j<yr$)r2r3qualified_name) python_class script_classs r _add_script_classr8Fs $0OL!6B\0023rcbt|dd}| t|}tj|dS)N_jit_override_qualname)getattr_get_python_classr2r)r6overrides r _get_script_classr>Ks4|%=tDH(2   |T 22rc.tj|dSr$)r3r)r5s r r<r<Rs    55rcTtjtjyr$)r2clearr3r+rr _clear_class_staterBVsr_jit_caching_layer_jit_function_overload_cachingctj|d}|r#|Dcgc]}tj|c}Sycc}wr$)rDr _python_cu find_function)key qual_names qual_names r _try_get_jit_cached_overloadsrKgs</33C>JEOP ((3PPQs=cP|Dcgc]}|jc}t|<ycc}wr$)r5rD)rH compiled_fnsfns r _set_jit_overload_cacherOos!GS*T2+<+<*T"3'*Ts#c~t|dddurytj|d}|rtj |Sy)N __disable_jit_function_caching__FT)r;rCrrFrG)rHrJs r _try_get_jit_cached_functionrRss@s6>$F"&&sD1I'' 22rcvt|tjjsJ|jt |<yr$) isinstancetorchjitScriptFunctionr5rC)rHr"s r _set_jit_function_cacherX}s- eUYY55 66 6#22sr)r*rweakreftypingrrUrr.r/r1_CCompilationUnitrFr2dicttype__annotations__r3strr8r>r<rBWeakKeyDictionaryrCrDrKrOrRrXr+rr rbs  > >XX % % ' /1d3ic*+0)+$sDI~&+C 361J0I0I0KG--K