diff --git a/altair/utils/_importers.py b/altair/utils/_importers.py index 14085ebcf..54d8ce3f4 100644 --- a/altair/utils/_importers.py +++ b/altair/utils/_importers.py @@ -12,6 +12,8 @@ def import_vegafusion() -> ModuleType: min_version = "1.5.0" try: + import vegafusion as vf # type: ignore + version = importlib_version("vegafusion") embed_version = importlib_version("vegafusion-python-embed") if version != embed_version or Version(version) < Version(min_version): @@ -23,7 +25,6 @@ def import_vegafusion() -> ModuleType: f" - vegafusion-python-embed=={embed_version}\n" ) raise RuntimeError(msg) - import vegafusion as vf # type: ignore return vf except ImportError as err: diff --git a/pyproject.toml b/pyproject.toml index f01a562bd..3e53c57fc 100644 --- a/pyproject.toml +++ b/pyproject.toml @@ -60,7 +60,7 @@ all = [ "pandas>=0.25.3", "numpy", "pyarrow>=11", - "vegafusion[embed]>=1.6.6", + "vegafusion[embed]>=1.6.6,<2", "anywidget>=0.9.0", "altair_tiles>=0.3.0" ]