JAMESPARK3's picture
Upload folder using huggingface_hub
1380717 verified
raw
history blame
816 Bytes
from typing import Final
from altair.utils._importers import import_vl_convert
from altair.utils.compiler import VegaLiteCompilerRegistry
ENTRY_POINT_GROUP: Final = "altair.vegalite.v5.vegalite_compiler"
vegalite_compilers = VegaLiteCompilerRegistry(entry_point_group=ENTRY_POINT_GROUP)
def vl_convert_compiler(vegalite_spec: dict) -> dict:
"""Vega-Lite to Vega compiler that uses vl-convert."""
from . import SCHEMA_VERSION
vlc = import_vl_convert()
# Compute vl-convert's vl_version string (of the form 'v5_8')
# from SCHEMA_VERSION (of the form 'v5.8.0')
vl_version = "_".join(SCHEMA_VERSION.split(".")[:2])
return vlc.vegalite_to_vega(vegalite_spec, vl_version=vl_version)
vegalite_compilers.register("vl-convert", vl_convert_compiler)
vegalite_compilers.enable("vl-convert")