diff --git a/mesa/visualization/__init__.py b/mesa/visualization/__init__.py index 529da658ee4..00c6f609793 100644 --- a/mesa/visualization/__init__.py +++ b/mesa/visualization/__init__.py @@ -12,10 +12,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", diff --git a/mesa/visualization/types.py b/mesa/visualization/types.py new file mode 100644 index 00000000000..6982f3a9807 --- /dev/null +++ b/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