a  d/ @sdZdgZddlZddlZddlZddlmZddlZddl m Z ddl m Z zddlZdZWneytdZYn0d d e jiiZd d e jiiZd d Zd dZeejddddejdddddejdddddddZdS)z8 Magic functions for rendering vega-lite specifications vegaliteN)magic_arguments)curried)v5TF vega-lite5cCsd|dust|tr|St|tjr2t||St|trDd|iStj d t |dd|SdS)z0Convert input data to data for use within schemaNurlzdata of type {} not recognized) stacklevel) isinstancedictpdZ DataFramerpipegetstrwarningswarnformattype)datadata_transformersrW/var/www/html/stable-diffusion-webui/venv/lib/python3.9/site-packages/altair/_magics.py _prepare_data&s  rcCs:t}|durtd||jvr0td||j|S)z+Get a variable from the notebook namespace.Nz[Magic command must be run within an IPython environemnt, in which get_ipython() is defined.z=argument '{}' does not match the name of any defined variable)IPythonZ get_ipython ValueErrorZuser_ns NameErrorr)nameiprrr _get_variable3s rr?zBlocal variablename of a pandas DataFrame to be used as the dataset)nargshelpz-vz --versionversionr)destdefaultz-jz--jsonjson store_true)r$actionc Cstt|}ddi}||j}|tdvs.Jtd|}td|}|jrXt|}nVtszt|}Wqtj y}zt d|WYd}~qd}~00nt j |t j d}|jdurt|j} t| ||d<||S)aHCell 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. rrrzH%%vegalite: spec is not valid JSON. Install pyyaml to parse spec as yamlN)Loaderr)rZparse_argstringrr# RENDERERS TRANSFORMERSr&loadsYAML_AVAILABLEJSONDecodeErrorryamlloadZ SafeLoaderrrr) linecellargsZexisting_versionsr#VegaLiterspecerrrrrrrCs,       )__doc____all__r&rrZ IPython.corerZpandasr ZtoolzrZaltair.vegaliterZ vegalite_v5r/r- ImportErrorr4r*rr+rrargumentrrrrrs@