diff --git a/mesa/experimental/__init__.py b/mesa/experimental/__init__.py index a48c498f6f9..5d396b5dc77 100644 --- a/mesa/experimental/__init__.py +++ b/mesa/experimental/__init__.py @@ -2,6 +2,12 @@ from mesa.experimental import cell_space -from .solara_viz import JupyterViz, Slider, SolaraViz, make_text +try: + from .solara_viz import JupyterViz, Slider, SolaraViz, make_text -__all__ = ["cell_space", "JupyterViz", "SolaraViz", "make_text", "Slider"] + __all__ = ["cell_space", "JupyterViz", "Slider", "SolaraViz", "make_text"] +except ImportError: + print( + "Could not import SolaraViz. If you need it, install with 'pip install --pre mesa[viz]'" + ) + __all__ = ["cell_space"]