fix: Do render_in_jupyter on Colab env

rht · tpike3 · commit 70b2902d10a7 · 2023-11-22T06:20:19.000-05:00
diff --git a/mesa/experimental/jupyter_viz.py b/mesa/experimental/jupyter_viz.py
@@ -152,7 +152,8 @@ def render_in_browser():
             on_grid_layout=set_grid_layout,
         )
 
-    if "ipykernel" in sys.argv[0]:
+    if ("ipykernel" in sys.argv[0]) or ("colab_kernel_launcher.py" in sys.argv[0]):
+        # When in Jupyter or Google Colab
         render_in_jupyter()
     else:
         render_in_browser()

Original file line number	Diff line number	Diff line change
`@@ -152,7 +152,8 @@ def render_in_browser():`
`152`	`152`	`on_grid_layout=set_grid_layout,`
`153`	`153`	`)`
`154`	`154`
`155`		`- if "ipykernel" in sys.argv[0]:`
	`155`	`+ if ("ipykernel" in sys.argv[0]) or ("colab_kernel_launcher.py" in sys.argv[0]):`
	`156`	`+ # When in Jupyter or Google Colab`
`156`	`157`	`render_in_jupyter()`
`157`	`158`	`else:`
`158`	`159`	`render_in_browser()`