YfvddlZddlZddlZddlmZej e5ddlmZm Z m Z m Z ddl m Z ddlmZddlmZe GddeZd ed dfd Zdddy#1swYyxYw) N)Any)Magicsline_cell_magic line_magic magics_class)scalene_profiler)ScaleneArguments)ScaleneParseArgsc`eZdZdZdededdfdZed dededdfd Ze d deddfd Z y) ScaleneMagicszHIPython (Jupyter) support for magics for Scalene (%scrun and %%scalene).argscodereturnNchddl}dt|jjjdz d}t |d5}|j |dddd|_tjjtjj||gdy#1swYRxYw) Nrz_ipython-input-z-profilezw+FT) is_jupyter) IPythonlen get_ipythonhistory_managerinput_hist_rawopenwritememoryrScaleneset_initialized run_profiler)selfr rrfilenametmpfiles V/var/www/highfloat_scraper/venv/lib/python3.12/site-packages/scalene/scalene_magics.pyrun_codezScaleneMagics.run_codes  )W-@-@-B-R-R-a-a)bcd)d(eemnHh% $ d# $DK  $ $ 4 4 6  $ $ 1 1xjT 2   $ $s B((B1linecellc|r5ddg|jt_tj\}}n t }|r|r|j |d|zyyy)zL%%scalene magic: see https://github.com/plasma-umass/scalene for usage info.scalene --ipython N)splitsysargvr parse_argsr r")rr#r$r _lefts r!r&zScaleneMagics.scalene&sX%{BTZZ\B 0 ; ; = u() dD4K0tctd|rYddg|jt_t j \}}|r"|j |dj|yyy)zI%scrun magic: see https://github.com/plasma-umass/scalene for usage info.z SCRUN MAGICr&r' N)printr)r*r+r r,r"join)rr#r lefts r!scrunzScaleneMagics.scrun5s[ - %{BTZZ\B/::< tMM$7r.)) __name__ __module__ __qualname____doc__r strr"rr&rr4r.r!r r skV !1     1 13 1 1  1  8c 84 8  8r.r iprcFtd|jttjt 5t dd5}|j}dddtj_ |tj_ dddtdjtjdtjdk(r8ttdjtjdyy#1swYxYw#1swYxYw)NLOADINGzscalene-usage.txtrr(a7Scalene extension successfully loaded. Note: Scalene currently only supports CPU+GPU profiling inside Jupyter notebooks. For full Scalene profiling, use the command line version. To profile in line mode, use `%scrun [options] statement`. To profile in cell mode, use `%%scalene [options]` followed by your code.darwinzNOTE: in Jupyter notebook on MacOS, Scalene cannot profile child processes. Do not run to try Scalene with multiprocessing in Jupyter Notebook.)r1register_magicsr contextlibsuppress Exceptionrreadr4r9r&r2textwrapwrapr*platform)r<usage usage_strs r!load_ipython_extensionrK?s i =)   + 6)3/ )5!JJL  )*3M   ',5M ! ! )  6  II N   <<8 # G  MMj  $ ) ) 6 6s# DD 2D D DD )rBr*rFtypingrrCrDIPython.core.magicrrrrr&rscalene.scalene_argumentsr scalene.scalene_parseargsr r rKr;r.r!rPs} Z#Q)::)8)8)8V34sQQQs 7A%%A.