o ^i@sUdZddlmZddlmZddlmZGdddeZGddde eZ Gd d d e Z Gd d d eZ Gd ddeZdZded<dZded<dZded<dS)zExceptions for nbclient.) annotations)Any) NotebookNodec@eZdZdZdS)CellControlSignalz A custom exception used to indicate that the exception is used for cell control actions (not the best model, but it's needed to cover existing behavior without major refactors). N__name__ __module__ __qualname____doc__r r N/home/ubuntu/hpml_nyu/venv/lib/python3.10/site-packages/nbclient/exceptions.pyr src@seZdZdZed d d Zd S) CellTimeoutErrorzS A custom exception to capture when a cell has timed out during execution. msgstrtimeoutintcellrreturncCsd|r&|jr&|jd}t|dkr|jn|ddd|dd}nd}|tj|||dS) z)Create an error from a timeout on a cell. Nz ... zCell contents not found.)rr cell_contents)sourcestripsplitlentimeout_err_msgformat)clsrrr src_by_linessrcr r r error_from_timeout_and_cells  z,CellTimeoutError.error_from_timeout_and_cellN)rrrrrrrr)rr r r classmethodr#r r r r rsrc@r)DeadKernelErrorzA dead kernel error.Nrr r r r r%)sr%c@r)CellExecutionCompletez Used as a control signal for cell execution across execute_cell and process_message function calls. Raised when all execution requests are completed and no further messages are expected from the kernel over zeromq channels. Nrr r r r r&/sr&csDeZdZdZdfdd Zdd d Zdd dZedddZZ S)CellExecutionErrorz Custom exception to propagate exceptions that are raised during notebook execution to the caller. This is mostly useful when using nbconvert as a library, since it allows to deal with failures gracefully. tracebackrenameevaluerNonecs"t|||_||_||_dS)zInitialize the error.N)super__init__r(r)r*)selfr(r)r* __class__r r r-Bs  zCellExecutionError.__init__ tuple[Any]cCst||j|j|jffS)zReduce implementation.)typer(r)r*r.r r r __reduce__IszCellExecutionError.__reduce__cCs|jr|jS|jd|jS)z Str repr.z: )r(r)r*r3r r r __str__MszCellExecutionError.__str__rrrdict[str, Any]cCsg}|jD]}|ddkr|tj|d|ddq|r+|dd|dd |}d |d gp9g}|tj|||d |d d |dddS)zvInstantiate from a code cell object and a message contents (message is either execute_reply or error) output_typestreamnametext)r9r:rz------------------rr()r stream_outputr(r)zr*)r)r*) outputsappendstream_output_msgrrstripinsertjoinget exec_err_msg)r rrstream_outputsoutputr<tbr r r from_cell_and_msgTs*       z$CellExecutionError.from_cell_and_msg)r(rr)rr*rrr+)rr1)rr)rrrr6rr') rr r r r-r4r5r$rH __classcell__r r r/r r':s  r'z----- {name} ----- {text}rr?zAn error occurred while executing the following cell: ------------------ {cell.source} ------------------ {stream_output} {traceback} rDzA cell timed out while it was being executed, after {timeout} seconds. The message was: {msg}. Here is a preview of the cell contents: ------------------- {cell_contents} ------------------- rN)r __future__rtypingrnbformatr Exceptionr TimeoutErrorr RuntimeErrorr%r&r'r?__annotations__rDrr r r r s     :