ukii PUddlmZddlZddlZddlZddlZddlmZddl Z ddl m Z ddl m Z ejddZiZded<ej$d ZGd d e j(Z dd Zej.j1d e j2dedyy)) annotationsN)Any) cli_debugger)corecnddl}ttt|jj dS)Nr.)web_pdbtuplemapint __version__split)r s Y/home/cdr/jupyterlab/.venv/lib/python3.12/site-packages/jax/_src/debugger/web_debugger.py_web_pdb_versionrs( s3++11#67 88zdict[tuple[str, int], Any] _web_consolesc~tjD]}|jtjyN)rvaluescloseclear)consoles r_close_debuggersr$s/%%'g MMOrc^eZdZUdZdZdZded< d d fd ZdZdZ d Z d Z xZ S) WebDebuggerzA web-based debugger.z(jdb) Fbool use_rawinputc||ftvr ddl}|j|||t||f<t||f}tj||_t ||||||y)Nr)stdinstdout completekey)rr WebConsoleweakrefproxy _debuggersuper__init__) selfframes thread_idr!hostportr _web_console __class__s rr'zWebDebugger.__init__0sk d|=(")"4"4T4"FmD$J t,L$]]40L GVYlr3linesr:s rget_current_frame_dataz"WebDebugger.get_current_frame_data;s&&(M%%H  EL'"))A-lI%99U#!5z 77??277??8#<= KGG$$X. %($##%//# rc|j}djdt|jj DS)Nr2c30K|]\}}|d|ywz = N.0keyvalues r z*WebDebugger.get_globals..Ws)A C %s5'A)r>rAsortedr<itemsr(r>s rrHzWebDebugger.get_globalsUsF&&(M 99A !6!6!?A AArc|j}djdt|jj DS)Nr2c30K|]\}}|d|ywrNrOrPs rrTz)WebDebugger.get_locals..]s)@ C %s5'@rU)r>rArVr=rWrXs rrIzWebDebugger.get_locals[sF&&(M 99@ !5!5!;!;!=>@ @@rc"|jSr)cmdloop)r(s rrunzWebDebugger.runas <<>r)tabi)r)!list[debugger_core.DebuggerFrame]r!strr+rar,r ) __name__ __module__ __qualname____doc__promptr__annotations__r'rKrHrIr] __classcell__)r.s@rrr+sO &,FJ . ./2 .?B .4A @ rrc :t||fi|jyr)rr])r)r*kwargss r run_debuggerrkds fi*6*..0rr web)returnztuple[int, ...])r)r`r*z int | Nonerjr) __future__ratexit functoolsimportlib.util importlibrCtypingrr#jax._src.debuggerrr debugger_corecacherrrgregisterr CliDebuggerrrkutil find_specregister_debuggerrOrrr}s#  *3 99 -/ ). 7,**7r1&1251  >>I&2!-!!%r:3r