refactor: Use vlc.get_vega_version()

#3633, vega/vl-convert#191
vega · Oct 7, 2024 · b0f1164 · b0f1164
1 parent 554410c
commit b0f1164
Show file tree

Hide file tree

Showing 2 changed files with 2 additions and 2 deletions.
diff --git a/tools/generate_schema_wrapper.py b/tools/generate_schema_wrapper.py
@@ -115,7 +115,6 @@ def vl_convert_to_vega_version(
 
 
 SCHEMA_VERSION: Final = "v5.20.1"
-VEGA_VERSION: Final[str] = vl_convert_to_vega_version()
 
 
 HEADER_COMMENT = """\
@@ -1282,7 +1281,7 @@ def main() -> None:
     copy_schemapi_util()
     vegalite_main(args.skip_download)
     write_expr_module(
-        VEGA_VERSION,
+        vlc.get_vega_version(),
         output=EXPR_FILE,
         header=HEADER_COMMENT,
     )

diff --git a/tools/schemapi/vega_expr.py b/tools/schemapi/vega_expr.py
@@ -960,6 +960,7 @@ def write_expr_module(version: str, output: Path, *, header: str) -> None:
     output
         Target path to write to.
     """
+    version = version if version.startswith("v") else f"v{version}"
     url = EXPRESSIONS_URL_TEMPLATE.format(version=version)
     content = (
         MODULE_PRE.format(