Simplify IPython detection

MilesCranmer · Feb 9, 2024 · 20646f9 · 20646f9
1 parent 2600822
commit 20646f9
Showing 1 changed file with 4 additions and 6 deletions.
diff --git a/pysr/julia_import.py b/pysr/julia_import.py
@@ -31,15 +31,13 @@
 
 
 def is_in_jupyter() -> bool:
-    try:
-        ipy = get_ipython().__class__.__name__  # type: ignore
-        return ipy == "ZMQInteractiveShell"
-    except NameError:
-        return False
+    return hasattr(__builtins__, "__IPYTHON__")
 
 
 if is_in_jupyter():
-    get_ipython().run_line_magic("load_ext", "julia.ipython")  # type: ignore
+    from IPython import get_ipython
+
+    get_ipython().run_line_magic("load_ext", "juliacall")
 
 
 from juliacall import Main as jl  # type: ignore