Automatically load ipython extension

MilesCranmer · Feb 9, 2024 · 2600822 · 2600822
1 parent d3c6385
commit 2600822
Showing 1 changed file with 13 additions and 0 deletions.
diff --git a/pysr/julia_import.py b/pysr/julia_import.py
@@ -29,6 +29,19 @@
 ):
     os.environ[k] = os.environ.get(k, default)
 
+
+def is_in_jupyter() -> bool:
+    try:
+        ipy = get_ipython().__class__.__name__  # type: ignore
+        return ipy == "ZMQInteractiveShell"
+    except NameError:
+        return False
+
+
+if is_in_jupyter():
+    get_ipython().run_line_magic("load_ext", "julia.ipython")  # type: ignore
+
+
 from juliacall import Main as jl  # type: ignore