BVhdZddlZddlZddlZddlZddlZddlZejdddgZdZ dZ dZ d Z d Z d Zy) z-A module to support operations on ipynb filesNCodeLine cell_numbercodecT|ddk(xr|dxr|ddjd S)z+Checks if the cell consists of Python code. cell_typersourcerz%% startswith)cells T/home/dcms/DCMS/lib/python3.12/site-packages/tensorflow/tools/compatibility/ipynb.py is_pythonr s? { v % 58n 5x.#..t446c4tdt|\}}|Dcgc]}|j}}tjdd5}|j dj ||\}} } } |r5|r3t||| jd} tj| |ntddj | z|} |j| ||}| }dddtjj| fScc}w#1swY3xYw) z;The function where we inject the support for ipynb upgrade.z,Extracting code lines from original notebookwF)delete z&Was not able to process the file: %s N)print _get_codertempfileNamedTemporaryFileupdate_string_pastajoin_update_notebooksplitjsondump SyntaxError _format_logshutilmovename) in_filename out_filenameupgraderraw_codenotebookcl raw_lines temp_fileprocessed_filenew_file_contentlogprocess_errors new_notebookfiles_processed report_texterrorss r process_filer3!s67 -(H!)*2rww*)*""3u5 $$TYYy%9;G:N$c>^%h&6&<&>> skip_magic('!ls -laF', ['%', '!', '?']) True TFr ) code_line magic_listmagics r skip_magicr8>s)eE"  rc.tjd|S)zChecks if a line was split with `\`. Args: code_line: A line of Python code Returns: If the line was split with `\` >>> skip_magic("!gcloud ml-engine models create ${MODEL} \\\n") True z\\\s*\n$)research)r5s r check_line_splitr<Ss ; **rc g}t|5}tj|}dddd}dD]}t|s|d}d}t |D]\}} t | gds|rd| z} t | }|r t | }|t|dz k(r#| jd r| jd d } |jt|| jjd d |dz }||fS#1swYxYw) z5Loads the ipynb file and returns a list of CodeLines.NrcellsrF)%!?###!!!r###===) openrloadr enumerater8r<lenendswithreplaceappendrrstrip) input_filer&in_filer' cell_indexr cell_lines is_line_splitline_idxr5s r rrcs1( J"7yy!H"*wd>jm!*:!6B (I i 1]*)+95- *95- J!+ +1C1CD1I''h7)  Z%%'//h? A B-B4Aoj?B 8 K""s DD ctj|}t|t|k(s)Jdjt|t|d}|dD]}t |st |Dcgc]\}}|j |k(r|}}}|Dcgc]}|| } }dj| jddjdd|d<|d z }|Scc}}wcc}w) z)Updates notebook, once migration is done.zCThe lengths of input and converted files are not the same: {} vs {}rr>rrBrrDrrC) copydeepcopyrHformatr rGrrrJ) original_notebookoriginal_raw_linesupdated_code_linesr/ code_cell_idxr idxr5applicable_linesnew_codes r rrs01,  C(:$; ;Js-.4F0GHJ ;-7# d T?#,,>"?Y  M 1  4DDC"3'DHDYYx(002>FF$DNQM   Es :C' C-)__doc__ collectionsrTrr:r r namedtuplerr r3r8r<rrrr rbsT4  !; ! !*}f.E F6.:* + *Zr