JupyterViz: Automatically deduce display name from model class

projectmesa · Jan 18, 2024 · 107d2a1 · 107d2a1
1 parent 877d552
commit 107d2a1
Showing 1 changed file with 4 additions and 1 deletion.
diff --git a/mesa/experimental/jupyter_viz.py b/mesa/experimental/jupyter_viz.py
@@ -17,7 +17,7 @@ def JupyterViz(
     model_class,
     model_params,
     measures=None,
-    name="Mesa Model",
+    name=None,
     agent_portrayal=None,
     space_drawer="default",
     play_interval=150,
@@ -35,6 +35,9 @@ def JupyterViz(
             specify `space_drawer=False`
         play_interval: play interval (default: 150)
     """
+    if name is None:
+        name = model_class.__name__
+
     current_step = solara.use_reactive(0)
 
     # 1. Set up model parameters