src/sage/__init__.py: Move monkey-patch of ExtensionFileLoader to src…

…/sage/cpython/__init__.py
sagemath · Sep 8, 2021 · 2e9ea56 · 2e9ea56
1 parent f77a98e
commit 2e9ea56
Show file tree

Hide file tree

Showing 2 changed files with 6 additions and 10 deletions.
diff --git a/src/sage/__init__.py b/src/sage/__init__.py
@@ -53,16 +53,6 @@ def isfunction(obj):
 inspect.isfunction = isfunction
 
 
-# Monkey-patch ExtensionFileLoader to allow IPython to find the sources
-# of Cython files. See https://trac.sagemath.org/ticket/24681
-try:
-    from importlib.machinery import ExtensionFileLoader
-except ImportError:
-    pass  # Python 2
-else:
-    del ExtensionFileLoader.get_source
-
-
 # Work around a Cygwin-specific bug caused by sqlite3; see
 # https://trac.sagemath.org/ticket/30157 and the docstring for
 # fix_for_ticket_30157

diff --git a/src/sage/cpython/__init__.py b/src/sage/cpython/__init__.py
@@ -7,3 +7,9 @@
 # to prevent the system zlib to be loaded instead of the Sage one.
 # See https://trac.sagemath.org/ticket/23122
 import zlib as _zlib
+
+
+# Monkey-patch ExtensionFileLoader to allow IPython to find the sources
+# of Cython files. See https://trac.sagemath.org/ticket/24681
+from importlib.machinery import ExtensionFileLoader as _ExtensionFileLoader
+del _ExtensionFileLoader.get_source