o hf @sdZddlmZddlZddlZddlmZddlmZddl m Z ddl m Z ddl mZd gZd d ejiiZd d ejiiZd d Zd$ddZe e jdddde jddddde jdddd d!d%d#d ZdS)&z7Magic functions for rendering vega-lite specifications.) annotationsN) find_spec)Any)magic_arguments)is_pandas_dataframe)v5vegalite vega-lite5cCsf|dus t|tr |St|r|}r||}|St|tr$d|iStjdt|ddd|S)z1Convert input data to data for use within schema.Nurlz data of type z not recognized) stacklevel) isinstancedictrgetstrwarningswarntype)datadata_transformersfuncrB/var/www/vscode/kcb/lib/python3.10/site-packages/altair/_magics.py _prepare_datas  rnamerreturnrcCsJddlm}|}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)rripmsgrrr _get_variable.s     r$r?zBlocal variablename of a pandas DataFrame to be used as the dataset)nargshelpz-vz --versionversionr)destdefaultz-jz--jsonjson store_true)r)actionvegalite_v5.VegaLitec Cstt|}ddi}||j}|tdvsJtd|}td|}|jr,t|}n,tdsLzt|}Wn tj yK}zd} t | |d}~wwddl } | j || j d}|jdurit|j} t| ||d <||S) 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_argstringrr( RENDERERS TRANSFORMERSr+loadsrJSONDecodeErrorr!r/load SafeLoaderrr$r) linecellargsexisting_versionsr(VegaLiterspecerrr#r/rrrrr?s.        )rrrr)rr.)__doc__ __future__rr+rimportlib.utilrtypingr IPython.corernarwhals.stable.v1.dependenciesraltair.vegaliter vegalite_v5__all__r<r2rr3rr$argumentrrrrrs: