adapt for new vegafusion and altair

fgcz · Jan 8, 2025 · f48a621 · f48a621
1 parent b530a82
commit f48a621
Show file tree

Hide file tree

Showing 3 changed files with 8 additions and 7 deletions.
diff --git a/pyproject.toml b/pyproject.toml
@@ -33,7 +33,7 @@ dependencies = [
     "statsmodels>=0.14.0",
     "tqdm>=4.66.1",
     "cyclopts",
-    "vegafusion[embed]>=1.6.5",
+    "vegafusion[embed]>=2.0.0",
     "xarray",
     # TODO unpin when https://github.com/Unidata/netcdf4-python/issues/1343 is resolved
     "netCDF4<=1.7.1",

diff --git a/src/depiction_targeted_preproc/workflow/exp/plot_compare_peak_density.py b/src/depiction_targeted_preproc/workflow/exp/plot_compare_peak_density.py
@@ -1,7 +1,8 @@
+from pathlib import Path
+
+import altair as alt
 import cyclopts
 import polars as pl
-import vegafusion
-from pathlib import Path
 
 from depiction_targeted_preproc.workflow.qc.plot_peak_density import plot_density_combined_full
 
@@ -14,7 +15,7 @@ def exp_plot_compare_peak_density(
     table_marker_distance_uncalib: Path,
     output_pdf: Path,
 ) -> None:
-    vegafusion.enable()
+    alt.data_transformers.enable("vegafusion")
 
     table = pl.concat(
         [

diff --git a/src/depiction_targeted_preproc/workflow/qc/plot_peak_density.py b/src/depiction_targeted_preproc/workflow/qc/plot_peak_density.py
@@ -1,9 +1,9 @@
+from pathlib import Path
+
 import altair as alt
 import cyclopts
 import polars as pl
-import vegafusion
 from KDEpy import FFTKDE
-from pathlib import Path
 
 from depiction_targeted_preproc.workflow.qc.plot_calibration_map import get_mass_groups
 
@@ -91,7 +91,7 @@ def qc_plot_peak_density(
     output_pdf: Path,
     grouped: bool = False,
 ) -> None:
-    vegafusion.enable()
+    alt.data_transformers.enable("vegafusion")
 
     table_calib = pl.read_parquet(table_marker_distances_calib)
     table_baseline = pl.read_parquet(table_marker_distances_baseline)