gf xdZddlmZddlZddlZddlmZddlmZddl m Z ddl m Z ddl mZd gZd d ej"iiZd d ej&iiZd Zdd Ze je j.ddde j.dddde j.ddddddZy)z7Magic functions for rendering vega-lite specifications.) annotationsN) find_spec)Any)magic_arguments)is_pandas_dataframe)v5vegalite vega-lite5c|t|tr|St|r|jx}r||}|St|trd|iSt j dt|dd|S)z1Convert input data to data for use within schema.urlz data of type z not recognized) stacklevel) isinstancedictrgetstrwarningswarntype)datadata_transformersfuncs C/var/www/openai/venv/lib/python3.12/site-packages/altair/_magics.py _prepare_datarss |z$- T "$((* *4 *:D D# t} d4j\AaP cddlm}|x}r.||jvrd|d}t||j|Sd}t |)z+Get a variable from the notebook namespace.r) get_ipythonz argument 'z1' does not match the name of any defined variablez[Magic command must be run within an IPython environment, in which get_ipython() is defined.)IPython.core.getipythonruser_ns NameError ValueError)nameripmsgs r _get_variabler&.s\3 ]r rzz !tf$UVCC. zz$ > orr?zBlocal variablename of a pandas DataFrame to be used as the dataset)nargshelpz-vz --versionversionr)destdefaultz-jz--jsonjson store_true)r+actionc2tjt|}ddi}||j}|tdvsJtd|}t d|}|j rt j|}nCtds t j|}n!ddl } | j|| j}|j$t|j} t!| ||d <||S#t j$r}d} t| |d}~wwxYw) aM Cell magic for displaying vega-lite visualizations in CoLab. %%vegalite [dataframe] [--json] [--version='v5'] Visualize the contents of the cell using Vega-Lite, optionally specifying a pandas DataFrame object to be used as the dataset. if --json is passed, then input is parsed as json rather than yaml. rr r yamlzH%%vegalite: spec is not valid JSON. Install pyyaml to parse spec as yamlNr)Loaderr)rparse_argstringr r* RENDERERS TRANSFORMERSr-loadsrJSONDecodeErrorr"r1load SafeLoaderrr&r) linecellargsexisting_versionsr*VegaLiterspecerrr%r1rs rr r ?s&  * *8T :Ds  -G i , ,, ,%g.H$[1': yyzz$ v  +::d#D yydooy6 yyTYY'$T+<=V D>## +7 S/s *  +sC00DDD)r#rreturnr)rAzvegalite_v5.VegaLite)__doc__ __future__rr-rimportlib.utilrtypingr IPython.corernarwhals.stable.v1.dependenciesraltair.vegaliter vegalite_v5__all__r>r4rr5rr&argumentr rrrMs=" $(?- , [ ! !   [ * *  "!  " M $ )TJ$vlK&LK #&r