o bi-@sddlZddlZddlZddlZddlZddlZddlmZddlm Z ddl Z ddl m Z eeZddZddZd d ZGd d d eZGd ddeZde edefddZde ede efddZGdddejZdS)N)closing)Optional)_ERROR_REPORT_TIMEOUTcCs`tttjtj}|d|tjtjd|dWdS1s)wYdS)N)r) rsocketAF_INET SOCK_STREAMbind setsockopt SOL_SOCKET SO_REUSEADDR getsockname)srJ/home/ubuntu/.local/lib/python3.10/site-packages/ray/air/_internal/util.pyfind_free_ports   $rcCs t|SN)npisnanvaluerrris_nans rcCst|pt|Sr)rrisinfrrrr is_nan_or_infsrc@seZdZdZdS)StartTracebackzMThese exceptions (and their tracebacks) can be skipped with `skip_exceptions`N)__name__ __module__ __qualname____doc__rrrrr srcs.eZdZdeddffdd ZddZZS)StartTracebackWithWorkerRank worker_rankreturnNcst||_dSr)super__init__r!)selfr! __class__rrr$'s  z%StartTracebackWithWorkerRank.__init__cCs|j|jffSr)r'r!)r%rrr __reduce__+sz'StartTracebackWithWorkerRank.__reduce__)rrrintr$r( __classcell__rrr&rr &sr excr"cCsbtttjdd}|r|St|trt|jSt | |j }t |dd}|r/t||_|S)a Skip all contained `StartTracebacks` to reduce traceback output. Returns a shallow copy of the exception with all `StartTracebacks` removed. If the RAY_AIR_FULL_TRACEBACKS environment variable is set, the original exception (not a copy) is returned. RAY_AIR_FULL_TRACEBACKS0 __cause__N) boolr)osenvironget isinstancerskip_exceptionsr.copywith_traceback __traceback__getattr)r+should_not_shortennew_exccauserrrr4/s    r4cCs|sdSt|ddS)Nr.)r8)r+rrrexception_causeKs r<cs@eZdZdZddZdefddZddZd fd d ZZ S) RunnerThreadz(Supervisor thread that runs your script.cOs*tjj|g|Ri|||_d|_dSr) threadingThreadr$ _error_queue_ret)r% error_queueargskwargsrrrr$Us zRunnerThread.__init__ecCs:z |jj|dtdWdStjytdYdSw)NT)blocktimeoutzRunner Thread was unable to report error to main function runner thread. This means a previous error was not processed. This should never happen.)r@putrqueueFullloggercriticalr%rErrr_propagate_exceptionZs z!RunnerThread._propagate_exceptionc Csz|j|ji|j|_WdStytdYdStyF}z|jdkr/tdn | |WYd}~dSWYd}~dSd}~wt y]}z | |WYd}~dSd}~ww)NzfThread runner raised StopIteration. Interpreting it as a signal to terminate the thread without error.rzuThread runner raised SystemExit with error code 0. Interpreting it as a signal to terminate the thread without error.) _target_args_kwargsrA StopIterationrKdebug SystemExitcoderN BaseExceptionrMrrrrunis$     zRunnerThread.runNcstt|||jSr)r#r=joinrA)r%rGr&rrrXszRunnerThread.joinr) rrrrr$rVrNrWrXr*rrr&rr=Rs r=)r5loggingr0rIrr> contextlibrtypingrnumpyrray.air.constantsr getLoggerrrKrrr Exceptionrr r4r<r?r=rrrrs&