fix: Do render_in_jupyter on Colab env

projectmesa · Nov 22, 2023 · 70b2902 · 70b2902
1 parent 51a1306
commit 70b2902
Showing 1 changed file with 2 additions and 1 deletion.
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()