Honor disabled space drawer when rendering in the browser

projectmesa · Dec 16, 2023 · a97a483 · a97a483
1 parent ca5e8f9
commit a97a483
Showing 1 changed file with 5 additions and 5 deletions.
diff --git a/mesa/experimental/jupyter_viz.py b/mesa/experimental/jupyter_viz.py
@@ -124,12 +124,12 @@ def render_in_jupyter():
                     make_plot(model, measure)
 
     def render_in_browser():
+        # if space drawer is disabled, do not include it
+        layout_types = [{"Space": "default"}] if space_drawer else []
+
         if measures:
-            layout_types = [{"Space": "default"}] + [
-                {"Measure": elem} for elem in range(len(measures))
-            ]
-        else:
-            layout_types = [{"Space": "default"}]
+            layout_types += [{"Measure": elem} for elem in range(len(measures))]
+
         grid_layout_initial = get_initial_grid_layout(layout_types=layout_types)
         grid_layout, set_grid_layout = solara.use_state(grid_layout_initial)