Adds Altair plot component and some other fixes

mesa/visualization/__init__.py

@@ -10,7 +10,7 @@
 )
 
 from .components import make_plot_component, make_space_component
-from .components.altair_components import make_space_altair
+from .components.altair_components import make_altair_plot_component, make_space_altair
 from .solara_viz import JupyterViz, SolaraViz
 from .user_param import Slider
 
@@ -19,6 +19,7 @@
     "Slider",
     "SolaraViz",
     "draw_space",
+    "make_altair_plot_component",
     "make_plot_component",
     "make_space_altair",
     "make_space_component",

mesa/visualization/components/__init__.py

@@ -4,7 +4,11 @@
 
 from collections.abc import Callable
 
-from .altair_components import SpaceAltair, make_altair_space
+from .altair_components import (
+    SpaceAltair,
+    make_altair_plot_component,
+    make_altair_space,
+)
 from .matplotlib_components import (
     SpaceMatplotlib,
     make_mpl_plot_component,
@@ -67,16 +71,13 @@ def make_plot_component(
         backend: the backend to use {"matplotlib", "altair"}
         plot_drawing_kwargs: additional keyword arguments to pass onto the backend specific function for making a plotting component
 
-    Notes:
-        altair plotting backend is not yet implemented and planned for mesa 3.1.
-
     Returns:
         function: A function that creates a plot component
     """
     if backend == "matplotlib":
         return make_mpl_plot_component(measure, post_process, **plot_drawing_kwargs)
     elif backend == "altair":
-        raise NotImplementedError("altair line plots are not yet implemented")
+        return make_altair_plot_component(measure, post_process, **plot_drawing_kwargs)
     else:
         raise ValueError(
             f"unknown backend {backend}, must be one of matplotlib, altair"