first commit

mesa/visualization/__init__.py

@@ -13,10 +13,14 @@
 from .components import make_plot_component, make_space_component
 from .components.altair_components import make_space_altair
 from .solara_viz import JupyterViz, SolaraViz
+from .types import HexGrid, Network, OrthogonalGrid
 from .user_param import Slider
 
 __all__ = [
+    "HexGrid",
     "JupyterViz",
+    "Network",
+    "OrthogonalGrid",
     "Slider",
     "SolaraViz",
     "draw_space",

mesa/visualization/types.py

@@ -0,0 +1,23 @@
+"""This module defines the types used in the visualization modules."""
+
+from mesa.experimental.cell_space import (
+    HexGrid as ExperimentalHexGrid,
+)
+from mesa.experimental.cell_space import (
+    Network as ExperimentalNetwork,
+)
+from mesa.experimental.cell_space import (
+    OrthogonalMooreGrid,
+    OrthogonalVonNeumannGrid,
+)
+from mesa.space import (
+    HexMultiGrid,
+    HexSingleGrid,
+    MultiGrid,
+    NetworkGrid,
+    SingleGrid,
+)
+
+OrthogonalGrid = SingleGrid | MultiGrid | OrthogonalMooreGrid | OrthogonalVonNeumannGrid
+HexGrid = HexSingleGrid | HexMultiGrid | ExperimentalHexGrid
+Network = NetworkGrid | ExperimentalNetwork