From 11f47181bd9f166231ef4fe6b963f31875fa0aa9 Mon Sep 17 00:00:00 2001
From: Jon Mease <jonmmease@gmail.com>
Date: Sat, 28 Sep 2024 19:48:16 -0400
Subject: [PATCH] v2: Remove altair functionality from vegafusion package
 (#510)

* Remove altair functionality from vegafusion package

* remove altair dep

* Install altair when testing
---
 .github/workflows/build_test.yml              |   6 +-
 pixi.lock                                     |  50 +++---
 pixi.toml                                     |   1 +
 python/vegafusion/setup.cfg                   |   1 -
 .../vegafusion/tests/test_context_manager.py  |  34 ----
 python/vegafusion/tests/test_row_limit.py     |  32 ----
 python/vegafusion/tests/test_save.py          |  59 -------
 .../vegafusion/tests/test_transformed_data.py |  10 +-
 python/vegafusion/vegafusion/__init__.py      | 111 -------------
 python/vegafusion/vegafusion/evaluation.py    |  13 --
 python/vegafusion/vegafusion/jupyter.py       |   4 -
 python/vegafusion/vegafusion/renderer.py      |  83 ----------
 python/vegafusion/vegafusion/save.py          | 152 ------------------
 python/vegafusion/vegafusion/transformer.py   |  60 -------
 14 files changed, 31 insertions(+), 585 deletions(-)
 delete mode 100644 python/vegafusion/tests/test_context_manager.py
 delete mode 100644 python/vegafusion/tests/test_row_limit.py
 delete mode 100644 python/vegafusion/tests/test_save.py
 delete mode 100644 python/vegafusion/vegafusion/evaluation.py
 delete mode 100644 python/vegafusion/vegafusion/jupyter.py
 delete mode 100644 python/vegafusion/vegafusion/renderer.py
 delete mode 100644 python/vegafusion/vegafusion/save.py

diff --git a/.github/workflows/build_test.yml b/.github/workflows/build_test.yml
index 458b30c2c..7c5520ead 100644
--- a/.github/workflows/build_test.yml
+++ b/.github/workflows/build_test.yml
@@ -299,7 +299,7 @@ jobs:
           python -m pip install polars-lts-cpu "duckdb>=1.0" "vl-convert-python>=1.0.1rc1" scikit-image "pandas>=2.2" jupytext voila anywidget ipywidgets chromedriver-binary-auto
 
           # Test dependencies
-          python -m pip install pytest vega-datasets scikit-image jupytext voila ipykernel anywidget ipywidgets selenium flaky tenacity chromedriver-binary-auto 
+          python -m pip install pytest altair vega-datasets scikit-image jupytext voila ipykernel anywidget ipywidgets selenium flaky tenacity chromedriver-binary-auto 
       - name: Test lazy imports
         working-directory: python/vegafusion/
         run: python checks/check_lazy_imports.py
@@ -347,7 +347,7 @@ jobs:
           python -m pip install polars "duckdb>=1.0" vl-convert-python "pandas>=2.2"
 
           # Test dependencies
-          python -m pip install pytest vega-datasets scikit-image
+          python -m pip install pytest altair vega-datasets scikit-image
 
       - name: Test vegafusion
         working-directory: python/vegafusion/
@@ -392,7 +392,7 @@ jobs:
           python -m pip install pyarrow==10.0 altair==5.1.2 polars[timezone] "duckdb>=1.0" vl-convert-python 
           
           # Test dependencies
-          python -m pip install pytest vega-datasets scikit-image
+          python -m pip install pytest altair vega-datasets scikit-image
       - name: Test vegafusion
         working-directory: python/vegafusion/
         run: pytest --ignore=tests/test_altair_mocks.py
diff --git a/pixi.lock b/pixi.lock
index 63c282373..1a7959574 100644
--- a/pixi.lock
+++ b/pixi.lock
@@ -816,7 +816,7 @@ environments:
       - conda: https://conda.anaconda.org/conda-forge/noarch/pkgutil-resolve-name-1.3.10-pyhd8ed1ab_0.tar.bz2
       - conda: https://conda.anaconda.org/conda-forge/noarch/platformdirs-3.10.0-pyhd8ed1ab_0.conda
       - conda: https://conda.anaconda.org/conda-forge/noarch/pluggy-1.2.0-pyhd8ed1ab_0.conda
-      - conda: https://conda.anaconda.org/conda-forge/osx-arm64/polars-0.18.15-py310had9acf8_0.conda
+      - conda: https://conda.anaconda.org/conda-forge/osx-arm64/polars-0.18.15-py310h49106b5_1.conda
       - conda: https://conda.anaconda.org/conda-forge/noarch/pooch-1.7.0-pyha770c72_3.conda
       - conda: https://conda.anaconda.org/conda-forge/noarch/prometheus_client-0.17.1-pyhd8ed1ab_0.conda
       - conda: https://conda.anaconda.org/conda-forge/noarch/prompt-toolkit-3.0.39-pyha770c72_0.conda
@@ -12051,12 +12051,10 @@ packages:
   depends:
   - libgcc-ng >=12
   - numpy >=1.16.0
-  - packaging *
+  - packaging
   - python >=3.10,<3.11.0a0
   - python_abi 3.10.* *_cp310
   - typing_extensions >=4.0.0
-  arch: x86_64
-  platform: linux
   license: MIT
   license_family: MIT
   size: 16057056
@@ -12064,44 +12062,42 @@ packages:
 - kind: conda
   name: polars
   version: 0.18.15
-  build: py310h95fa17d_1
+  build: py310h49106b5_1
   build_number: 1
-  subdir: osx-64
-  url: https://conda.anaconda.org/conda-forge/osx-64/polars-0.18.15-py310h95fa17d_1.conda
-  sha256: 423ebf36060323f5f145b23437dbbdf7c5735d1f5c536adcc463d2434b1812f1
-  md5: ca50abd3aa922d6d8d8f1a3ff0dda332
+  subdir: osx-arm64
+  url: https://conda.anaconda.org/conda-forge/osx-arm64/polars-0.18.15-py310h49106b5_1.conda
+  sha256: ebc61b1e954585f492e777cd5d2caa06c2e0e28c0be364ab0c412294b72fe13b
+  md5: 51481d96d1b4c863662e81d075ad5d79
   depends:
   - numpy >=1.16.0
-  - packaging *
+  - packaging
   - python >=3.10,<3.11.0a0
+  - python >=3.10,<3.11.0a0 *_cpython
   - python_abi 3.10.* *_cp310
   - typing_extensions >=4.0.0
-  arch: x86_64
-  platform: osx
   license: MIT
   license_family: MIT
-  size: 13970392
-  timestamp: 1692690680600
+  size: 12963300
+  timestamp: 1692690752488
 - kind: conda
   name: polars
   version: 0.18.15
-  build: py310had9acf8_0
-  subdir: osx-arm64
-  url: https://conda.anaconda.org/conda-forge/osx-arm64/polars-0.18.15-py310had9acf8_0.conda
-  sha256: 9d50cec48ac801a5ec4b44fb3c00d761ce55d1e1af479cef1d9cff3dd7c66dfa
-  md5: 9fb570a944037614daba1b125648588d
+  build: py310h95fa17d_1
+  build_number: 1
+  subdir: osx-64
+  url: https://conda.anaconda.org/conda-forge/osx-64/polars-0.18.15-py310h95fa17d_1.conda
+  sha256: 423ebf36060323f5f145b23437dbbdf7c5735d1f5c536adcc463d2434b1812f1
+  md5: ca50abd3aa922d6d8d8f1a3ff0dda332
   depends:
   - numpy >=1.16.0
-  - python >=3.10,<3.11.0a0 *_cpython
+  - packaging
+  - python >=3.10,<3.11.0a0
   - python_abi 3.10.* *_cp310
   - typing_extensions >=4.0.0
-  - packaging *
-  arch: aarch64
-  platform: osx
   license: MIT
   license_family: MIT
-  size: 12977174
-  timestamp: 1692100866706
+  size: 13970392
+  timestamp: 1692690680600
 - kind: conda
   name: polars
   version: 0.18.15
@@ -12113,15 +12109,13 @@ packages:
   md5: 14eed67eba855b1ddcffb7f89586875e
   depends:
   - numpy >=1.16.0
-  - packaging *
+  - packaging
   - python >=3.10,<3.11.0a0
   - python_abi 3.10.* *_cp310
   - typing_extensions >=4.0.0
   - ucrt >=10.0.20348.0
   - vc >=14.2,<15
   - vc14_runtime >=14.29.30139
-  arch: x86_64
-  platform: win
   license: MIT
   license_family: MIT
   size: 14328233
diff --git a/pixi.toml b/pixi.toml
index c875a7307..2a369c179 100644
--- a/pixi.toml
+++ b/pixi.toml
@@ -139,6 +139,7 @@ ipywidgets = "8.1.0.*"
 rust = "1.80.*"
 vl-convert-python = "1.6.*"
 anywidget = ">=0.9.6,<0.10"
+polars = "1.8.*"
 
 [target.osx-arm64.build-dependencies]
 # These dependencies are for building node canvas from source on apple silicon
diff --git a/python/vegafusion/setup.cfg b/python/vegafusion/setup.cfg
index 8da2b7153..4e5572ef1 100644
--- a/python/vegafusion/setup.cfg
+++ b/python/vegafusion/setup.cfg
@@ -27,7 +27,6 @@ packages = find:
 python_requires = >=3.7
 include_package_data = True
 install_requires = 
-	altair>=5.2.0
 	pyarrow>=5
 	pandas
 	psutil
diff --git a/python/vegafusion/tests/test_context_manager.py b/python/vegafusion/tests/test_context_manager.py
deleted file mode 100644
index cbc8d4325..000000000
--- a/python/vegafusion/tests/test_context_manager.py
+++ /dev/null
@@ -1,34 +0,0 @@
-import altair as alt
-import vegafusion as vf
-
-
-def test_mime_enabler_context_manager():
-    alt.data_transformers.enable("json")
-    alt.renderers.enable("mimetype")
-
-    assert alt.data_transformers.active == "json"
-    assert alt.renderers.active == "mimetype"
-
-    with vf.disable():
-        assert alt.data_transformers.active == "default"
-        assert alt.renderers.active == "default"
-
-    assert alt.data_transformers.active == "json"
-    assert alt.renderers.active == "mimetype"
-
-    ctx = vf.disable()
-    assert alt.data_transformers.active == "default"
-    assert alt.renderers.active == "default"
-    assert repr(ctx) == "vegafusion.disable()"
-
-    with vf.enable():
-        assert alt.data_transformers.active == "vegafusion-inline"
-        assert alt.renderers.active == "vegafusion-mime"
-
-    assert alt.data_transformers.active == "default"
-    assert alt.renderers.active == "default"
-
-    ctx = vf.enable()
-    assert alt.data_transformers.active == "vegafusion-inline"
-    assert alt.renderers.active == "vegafusion-mime"
-    assert repr(ctx) == "vegafusion.enable(mimetype='html', row_limit=10000, embed_options={})"
diff --git a/python/vegafusion/tests/test_row_limit.py b/python/vegafusion/tests/test_row_limit.py
deleted file mode 100644
index f9bbbd1a5..000000000
--- a/python/vegafusion/tests/test_row_limit.py
+++ /dev/null
@@ -1,32 +0,0 @@
-import pytest
-import vegafusion as vf
-import altair as alt
-from altair.utils.execeval import eval_block
-from pathlib import Path
-here = Path(__file__).parent
-altair_mocks_dir = here / "altair_mocks"
-
-
-def test_row_limit():
-    mock_path = altair_mocks_dir / "scatter" / "bubble_plot" / "mock.py"
-    mock_src = mock_path.read_text("utf8")
-    chart = eval_block(mock_src)
-
-    # Dataset has 406 rows (before filtering out nulls) and this chart is not aggregated.
-    # Limit of 500 rows should be fine
-    with vf.enable(row_limit=500):
-        chart._repr_mimebundle_()
-
-    # Limit of 300 rows should raise RowLimitError
-    with vf.enable(row_limit=300):
-        with pytest.raises(vf.RowLimitError):
-            chart._repr_mimebundle_()
-
-    # Adding an aggregation should allow row limit to be much lower
-    chart = chart.encode(
-        alt.X("Horsepower", bin=True),
-        alt.Y("Miles_per_Gallon", bin=True),
-        alt.Size("count()")
-    )
-    with vf.enable(row_limit=50):
-        chart._repr_mimebundle_()
diff --git a/python/vegafusion/tests/test_save.py b/python/vegafusion/tests/test_save.py
deleted file mode 100644
index 1c639a2d0..000000000
--- a/python/vegafusion/tests/test_save.py
+++ /dev/null
@@ -1,59 +0,0 @@
-import altair as alt
-from vega_datasets import data
-import vegafusion as vf
-from io import StringIO, BytesIO
-import json
-
-
-def make_histogram():
-    source = data.movies.url
-    return alt.Chart(source).mark_bar().encode(
-        alt.X("IMDB_Rating:Q", bin=True),
-        y='count()',
-    )
-
-
-def test_save_html():
-    chart = make_histogram()
-    f = StringIO()
-    vf.save_html(chart, f)
-    html = f.getvalue().strip()
-    assert html.startswith("<!DOCTYPE html>")
-    assert "https://cdn.jsdelivr.net/npm/vega@5" in html
-    assert "https://cdn.jsdelivr.net/npm/vega-embed@6" in html
-
-
-# We can't test inline HTML until after a new release of Altair Viewer
-# def test_save_html_inline():
-#     chart = make_histogram()
-#     f = StringIO()
-#     vf.save_html(chart, f, inline=True)
-#     html = f.getvalue().strip()
-#     assert html.startswith("<!DOCTYPE html>")
-#     assert "<script type=\"text/javascript\">" in html
-#     assert "https://cdn.jsdelivr.net/npm/vega@5" not in html
-#     assert "https://cdn.jsdelivr.net/npm/vega-embed@6" not in html
-
-
-def test_save_vega():
-    chart = make_histogram()
-    f = StringIO()
-    vf.save_vega(chart, f)
-    vega = json.loads(f.getvalue())
-    assert vega["$schema"] == "https://vega.github.io/schema/vega/v5.json"
-
-
-def test_save_svg():
-    chart = make_histogram()
-    f = StringIO()
-    vf.save_svg(chart, f)
-    svg = f.getvalue()
-    assert svg.startswith("<svg")
-
-
-def test_save_png():
-    chart = make_histogram()
-    f = BytesIO()
-    vf.save_png(chart, f)
-    png = f.getvalue()
-    assert png.startswith(b'\x89PNG')
diff --git a/python/vegafusion/tests/test_transformed_data.py b/python/vegafusion/tests/test_transformed_data.py
index a63d34cb0..81cb1e97b 100644
--- a/python/vegafusion/tests/test_transformed_data.py
+++ b/python/vegafusion/tests/test_transformed_data.py
@@ -117,7 +117,7 @@ def test_transformed_data_for_mock(mock_name, expected_len, expected_cols, conne
     mock_path = altair_mocks_dir / mock_name / "mock.py"
     mock_src = mock_path.read_text("utf8")
     chart = eval_block(mock_src)
-    df = vf.transformed_data(chart)
+    df = chart.transformed_data()
 
     # Check that a DataFrame was returned
     assert isinstance(df, pd.DataFrame)
@@ -182,7 +182,7 @@ def test_multi_transformed_data_for_mock(mock_name, expected_lens, all_expected_
     mock_path = altair_mocks_dir / mock_name / "mock.py"
     mock_src = mock_path.read_text("utf8")
     chart = eval_block(mock_src)
-    dfs = vf.transformed_data(chart)
+    dfs = chart.transformed_data()
 
     for df, expected_len, expected_cols in zip(dfs, expected_lens, all_expected_cols):
         # Check that a DataFrame was returned
@@ -211,7 +211,7 @@ def test_transformed_data_exclude():
     )
 
     chart = (bar + rule + some_annotation).properties(width=600)
-    datasets = vf.transformed_data(chart, exclude=["some_annotation"])
+    datasets = chart.transformed_data(exclude=["some_annotation"])
 
     assert len(datasets) == 2
     assert len(datasets[0]) == 52
@@ -234,7 +234,7 @@ def test_gh_286():
         y='count():Q',
         color='weather:N'
     )
-    transformed = vf.transformed_data(chart)
+    transformed = chart.transformed_data()
     assert isinstance(transformed, pl.DataFrame)
     assert len(transformed) == 53
 
@@ -251,6 +251,6 @@ def test_categorical_columns(connection):
     chart = alt.Chart(df).mark_bar().encode(
         alt.X("categorical:N"), alt.Y("sum(a):Q")
     )
-    transformed = vf.transformed_data(chart)
+    transformed = chart.transformed_data()
     expected = pd.DataFrame({"categorical": ["A", "BB"], "sum_a": [7, 8]})
     pd.testing.assert_frame_equal(transformed, expected)
\ No newline at end of file
diff --git a/python/vegafusion/vegafusion/__init__.py b/python/vegafusion/vegafusion/__init__.py
index 880d76f2e..b18949df8 100644
--- a/python/vegafusion/vegafusion/__init__.py
+++ b/python/vegafusion/vegafusion/__init__.py
@@ -1,123 +1,12 @@
 from .runtime import runtime
 from .transformer import to_feather, get_inline_datasets_for_spec
-from .renderer import RowLimitError
 from .local_tz import set_local_tz, get_local_tz
-from .evaluation import transformed_data
-from .save import save_html, save_vega, save_png, save_svg
-from . import renderer
 
 
 from ._version import __version__
 from .utils import RendererTransformerEnabler
 
-# Import optional subpackages
-try:
-    import vegafusion.jupyter
-except ImportError:
-    pass
-
 try:
     import vegafusion.embed
 except ImportError:
     pass
-
-
-def altair_vl_version(vl_convert=False):
-    """
-    Get Altair's preferred Vega-Lite version
-
-    :param vl_convert: If True, return a version string compatible with vl_convert
-        (e.g. v4_17 rather than 4.17.0)
-    :return: str with Vega-Lite version
-    """
-    from altair import SCHEMA_VERSION
-
-    if vl_convert:
-        # Compute VlConvert's vl_version string (of the form 'v5_2')
-        # from SCHEMA_VERSION (of the form 'v5.2.0')
-        return "_".join(SCHEMA_VERSION.split(".")[:2])
-    else:
-        # Return full version without leading v
-        return SCHEMA_VERSION.lstrip("v")
-
-
-def enable(mimetype="html", row_limit=10000, embed_options=None):
-    """
-    Enable the VegaFusion mime renderer so that all Altair charts
-    are displayed using VegaFusion.
-
-    :param mimetype: Mime type. One of:
-        - "html" (default)
-        - "vega"
-        - "svg"
-        - "png": Note, the PNG renderer can be quite slow for charts with lots of marks
-    :param row_limit: Maximum number of rows (after transforms are applied) that may be
-        included in the Vega specifications that will be displayed. An error will
-        be raised if the limit is exceeded. None for unlimited.
-    :param embed_options: dict (optional)
-        Dictionary of options to pass to the vega-embed. Default
-        entry is {'mode': 'vega'}.
-    """
-    import altair as alt
-    embed_options = embed_options if embed_options is not None else {}
-    return RendererTransformerEnabler(
-        renderer_ctx=alt.renderers.enable(
-            'vegafusion-mime', mimetype=mimetype, row_limit=row_limit, embed_options=embed_options
-        ),
-        data_transformer_ctx=alt.data_transformers.enable('vegafusion-inline'),
-        repr_str=(
-            "vegafusion.enable("
-            f"mimetype={repr(mimetype)}, row_limit={row_limit}, embed_options={repr(embed_options)}"
-            ")"
-        )
-    )
-
-
-def enable_widget(
-    debounce_wait=30,
-    debounce_max_wait=60,
-    data_dir="_vegafusion_data",
-    download_source_link=None,
-):
-    """
-    Enable the VegaFusion widget renderer so that all charts are displayed
-    using VegaFusion.
-
-    This isn't necessary in order to use the VegaFusionWidget directly
-
-    :param debounce_wait: Debounce wait period for interactions (milliseconds)
-    :param debounce_max_wait: Max debounce wait for interactions (milliseconds)
-    :param download_source_link: URL of notebooks source code for inclusion
-        in dropdown menu
-    :param data_dir: Path that temporary feather files will be written to
-    """
-    from vegafusion.jupyter import enable
-    return enable(
-        download_source_link=download_source_link,
-        debounce_wait=debounce_wait,
-        debounce_max_wait=debounce_max_wait,
-        data_dir=data_dir
-    )
-
-
-def disable():
-    """
-    Disable the VegaFusion data transformers and renderers so that Charts
-    are not displayed using VegaFusion
-
-    Equivalent to
-
-    ```python
-    import altair as alt
-    alt.renderers.enable('default')
-    alt.data_transformers.enable('default')
-    ```
-
-    This does not affect the behavior of VegaFusionWidget
-    """
-    import altair as alt
-    return RendererTransformerEnabler(
-        renderer_ctx=alt.renderers.enable('default'),
-        data_transformer_ctx=alt.data_transformers.enable('default'),
-        repr_str="vegafusion.disable()"
-    )
diff --git a/python/vegafusion/vegafusion/evaluation.py b/python/vegafusion/vegafusion/evaluation.py
deleted file mode 100644
index ee970bcdf..000000000
--- a/python/vegafusion/vegafusion/evaluation.py
+++ /dev/null
@@ -1,13 +0,0 @@
-def transformed_data(chart, row_limit=None, exclude=None):
-    """
-    Evaluate the transform associated with a Chart and return the transformed
-    data as one or more DataFrames
-
-    :param chart: altair Chart, LayerChart, HConcatChart, VConcatChart, or ConcatChart
-    :param row_limit: Maximum number of rows to return. None (default) for unlimited
-    :param exclude: list of mark names to exclude.
-    :return:
-        - DataFrame of transformed data for Chart input
-        - list of DataFrames for LayerChart, HConcatChart, VConcatChart, ConcatChart
-    """
-    return chart.transformed_data(row_limit=row_limit, exclude=exclude)
diff --git a/python/vegafusion/vegafusion/jupyter.py b/python/vegafusion/vegafusion/jupyter.py
deleted file mode 100644
index 85da6011c..000000000
--- a/python/vegafusion/vegafusion/jupyter.py
+++ /dev/null
@@ -1,4 +0,0 @@
-try:
-    from vegafusion_jupyter import *
-except ImportError:
-    raise ImportError("Please install the vegafusion-jupyter package")
diff --git a/python/vegafusion/vegafusion/renderer.py b/python/vegafusion/vegafusion/renderer.py
deleted file mode 100644
index 661812102..000000000
--- a/python/vegafusion/vegafusion/renderer.py
+++ /dev/null
@@ -1,83 +0,0 @@
-from uuid import uuid4
-
-class RowLimitError(ValueError):
-    def __init__(self, row_limit):
-        self.row_limit = row_limit
-
-    def __str__(self):
-        return (
-            f"Limit of {self.row_limit} rows was exceeded.\n"
-            "Either introduce an aggregation to reduce the number of rows sent to the client or\n"
-            "increase the row_limit argument to the vegafusion.enable() function"
-        )
-
-
-def vegafusion_mime_renderer(spec, mimetype="html", row_limit=None, embed_options=None):
-    return spec_to_mime_bundle(
-        spec,
-        mimetype=mimetype,
-        row_limit=row_limit,
-        embed_options=embed_options
-    )
-
-
-def spec_to_mime_bundle(
-        spec,
-        mimetype="html",
-        row_limit=None,
-        embed_options=None,
-        html_template="universal",
-        full_html=False,
-        scale=1,
-):
-    from altair import vegalite_compilers
-    from . import transformer, runtime, local_tz, altair_vl_version
-    vega_spec = vegalite_compilers.get()(spec)
-
-    inline_datasets = transformer.get_inline_datasets_for_spec(vega_spec)
-    tx_vega_spec, warnings = runtime.pre_transform_spec(
-        vega_spec,
-        local_tz.get_local_tz(),
-        row_limit=row_limit,
-        inline_datasets=inline_datasets
-    )
-
-    for warning in warnings:
-        if warning.get("type", "") == "RowLimitExceeded":
-            raise RowLimitError(row_limit)
-
-    # Handle default embed options
-    embed_options = embed_options or {}
-    embed_options = dict({"mode": "vega"}, **embed_options)
-
-    if mimetype == "vega":
-        vega_mimetype = "application/vnd.vega.v5+json"
-        return (
-            {vega_mimetype: tx_vega_spec},
-            {vega_mimetype: {"embed_options": embed_options}}
-        )
-    elif mimetype == "html":
-        from altair.utils.html import spec_to_html
-        output_div = f"altair-viz-{uuid4().hex}"
-        html = spec_to_html(
-            tx_vega_spec,
-            mode="vega",
-            vega_version="5",
-            vegalite_version=altair_vl_version(),
-            vegaembed_version="6",
-            fullhtml=full_html,
-            output_div=output_div,
-            template=html_template,
-            embed_options=embed_options
-        )
-        return {"text/html": html}
-    elif mimetype == "svg":
-        import vl_convert as vlc
-        svg = vlc.vega_to_svg(tx_vega_spec)
-        return {"image/svg+xml": svg}
-    elif mimetype == "png":
-        import vl_convert as vlc
-        png = vlc.vega_to_png(tx_vega_spec, scale=scale)
-        return {"image/png": png}
-    else:
-        raise ValueError(f"Unsupported mimetype: {mimetype}")
diff --git a/python/vegafusion/vegafusion/save.py b/python/vegafusion/vegafusion/save.py
deleted file mode 100644
index 5120045fc..000000000
--- a/python/vegafusion/vegafusion/save.py
+++ /dev/null
@@ -1,152 +0,0 @@
-import json
-import pathlib
-from .renderer import spec_to_mime_bundle
-
-
-def save_html(
-    chart,
-    file,
-    embed_options=None,
-    inline=False,
-    full_html=True,
-):
-    """
-    Save an Altair Chart to an HTML file after pre-applying data transformations
-    and removing unused columns
-
-    :param chart: alt.Chart
-        The Altair chart to save
-    :param file: str, pathlib.Path, or file-like object
-        The file path to write the HTML file to, or a file-like object to write to
-    :param embed_options: dict (default None)
-        Dictionary of options to pass to vega-embed
-    :param inline: boolean (default False)
-        If False (default), the required JavaScript libraries are loaded from a
-        CDN location. This results in a smaller file, but an internet connection
-        is required to view the resulting HTML file.
-
-        If True, all required JavaScript libraries are inlined.
-        This results in a larger file, but no internet connection is required to view.
-        inline=True requires Altair 5 and the altair_viewer package
-    :param full_html: boolean (default True)
-        If True, then a full html page is written. If False, then
-        an HTML snippet that can be embedded into an HTML page is written.
-    """
-    from . import enable
-
-    with enable():
-        vegalite_spec = chart.to_dict()
-
-    bundle = spec_to_mime_bundle(
-        spec=vegalite_spec,
-        mimetype="html",
-        embed_options=embed_options,
-        html_template="inline" if inline else "standard",
-        full_html=full_html,
-    )
-
-    html = bundle["text/html"]
-    write_file_or_filename(file, html)
-
-
-def save_vega(
-    chart,
-    file,
-    pretty=True,
-):
-    """
-    Save an Altair Chart to a Vega JSON file after pre-applying data transformations
-    and removing unused columns
-
-    :param chart: alt.Chart
-        The Altair chart to save
-    :param file: str, pathlib.Path, or file-like object
-        The file path to write the Vega JSON file to, or a file-like object to write to
-    :param pretty: boolean (default True)
-        If True, pretty-print the resulting JSON file. If False, write the smallest file possible
-    """
-    from . import enable
-
-    with enable():
-        vegalite_spec = chart.to_dict()
-
-    (bundle, _) = spec_to_mime_bundle(
-        spec=vegalite_spec,
-        mimetype="vega",
-    )
-
-    vega_dict = bundle["application/vnd.vega.v5+json"]
-    if pretty:
-        vega_str = json.dumps(vega_dict, indent=2)
-    else:
-        vega_str = json.dumps(vega_dict)
-
-    write_file_or_filename(file, vega_str)
-
-
-def save_png(
-    chart,
-    file,
-    scale=1,
-):
-    """
-    Save an Altair Chart to a static PNG image file after pre-applying data transformations
-    and removing unused columns
-
-    :param chart: alt.Chart
-        The Altair chart to save
-    :param file: str, pathlib.Path, or file-like object
-        The file path to write the PNG file to, or a file-like object to write to
-    :param scale: float (default 1)
-        Image scale factor
-    """
-    from . import enable
-
-    with enable():
-        vegalite_spec = chart.to_dict()
-
-    bundle = spec_to_mime_bundle(
-        spec=vegalite_spec,
-        mimetype="png",
-        scale=scale,
-    )
-
-    png = bundle["image/png"]
-    write_file_or_filename(file, png, "wb")
-
-
-def save_svg(
-    chart,
-    file,
-):
-    """
-    Save an Altair Chart to a static SVG image file after pre-applying data transformations
-    and removing unused columns
-
-    :param chart: alt.Chart
-        The Altair chart to save
-    :param file: str, pathlib.Path, or file-like object
-        The file path to write the SVG file to, or a file-like object to write to
-    """
-    from . import enable
-
-    with enable():
-        vegalite_spec = chart.to_dict()
-
-    bundle = spec_to_mime_bundle(
-        spec=vegalite_spec,
-        mimetype="svg",
-    )
-
-    svg = bundle["image/svg+xml"]
-    write_file_or_filename(file, svg, "w")
-
-
-def write_file_or_filename(file, content, mode="w"):
-    """Write content to file, whether file is a string, a pathlib Path or a
-    file-like object"""
-    if isinstance(file, str) or isinstance(file, pathlib.PurePath):
-        with open(file, mode) as f:
-            f.write(content)
-    else:
-        file.write(content)
diff --git a/python/vegafusion/vegafusion/transformer.py b/python/vegafusion/vegafusion/transformer.py
index 0bcd1d154..f307b3dc0 100644
--- a/python/vegafusion/vegafusion/transformer.py
+++ b/python/vegafusion/vegafusion/transformer.py
@@ -132,51 +132,6 @@ def to_feather(data, file):
             f.write(file_bytes)
 
 
-def feather_transformer(data, data_dir="_vegafusion_data"):
-    from vegafusion import runtime
-    import altair as alt
-
-    if "vegafusion" not in alt.renderers.active:
-        # Use default transformer if a vegafusion renderer is not active
-        return alt.default_data_transformer(data)
-    elif has_geo_interface(data):
-        # Use default transformer for geo interface objects
-        # (e.g. a geopandas GeoDataFrame)
-        return alt.default_data_transformer(data)
-    elif is_dataframe_like(data):
-        if runtime.using_grpc:
-            raise ValueError(
-                "DataFrame data sources are not yet supported by the gRPC runtime"
-            )
-
-        bytes_buffer = io.BytesIO()
-        to_feather(data, bytes_buffer)
-        file_bytes = bytes_buffer.getvalue()
-
-        # Hash bytes to generate unique file name
-        hasher = sha1()
-        hasher.update(file_bytes)
-        hashstr = hasher.hexdigest()
-        fname = f"vegafusion-{hashstr}.feather"
-
-        # Check if file already exists
-        tmp_dir = pathlib.Path(data_dir) / "tmp"
-        os.makedirs(tmp_dir, exist_ok=True)
-        path = pathlib.Path(data_dir) / fname
-        if not path.is_file():
-            # Write to temporary file then move (os.replace) to final destination. This is more resistant
-            # to race conditions
-            with NamedTemporaryFile(dir=tmp_dir, delete=False) as tmp_file:
-                tmp_file.write(file_bytes)
-                tmp_name = tmp_file.name
-
-            os.replace(tmp_name, path)
-
-        return {"url": path.as_posix()}
-    else:
-        # Use default transformer if we don't recognize data
-        return alt.default_data_transformer(data)
-
 
 def get_inline_dataset_names(vega_spec):
     """
@@ -218,21 +173,6 @@ def get_inline_datasets_for_spec(vega_spec):
     return datasets
 
 
-def inline_data_transformer(data):
-    import altair as alt
-    if has_geo_interface(data):
-        # Use default transformer for geo interface objects
-        # # (e.g. a geopandas GeoDataFrame)
-        return alt.default_data_transformer(data)
-    elif is_dataframe_like(data):
-        table_name = f"table_{uuid.uuid4()}".replace("-", "_")
-        __inline_tables[table_name] = data
-        return {"url": DATASET_PREFIXES[0] + table_name}
-    else:
-        # Use default transformer if we don't recognize data
-        return alt.default_data_transformer(data)
-
-
 def is_dataframe_like(data):
     pa = sys.modules.get("pyarrow")
     pd = sys.modules.get("pandas")