restored sys modules after import

anjsudh · Dec 25, 2018 · 5933d5e · 5933d5e
1 parent 435923e
commit 5933d5e
Showing 1 changed file with 4 additions and 1 deletion.
diff --git a/pandas/core/computation/check.py b/pandas/core/computation/check.py
@@ -1,3 +1,4 @@
+import sys
 from distutils.version import LooseVersion
 import warnings
 
@@ -6,10 +7,12 @@
 _NUMEXPR_VERSION = None
 
 try:
+    modules = sys.modules.copy()
     import numexpr as ne
     ver = LooseVersion(ne.__version__)
     _NUMEXPR_INSTALLED = ver >= LooseVersion(_MIN_NUMEXPR_VERSION)
     _NUMEXPR_VERSION = ver
+    sys.modules = modules
 
     if not _NUMEXPR_INSTALLED:
         warnings.warn(
@@ -21,4 +24,4 @@
 except ImportError:  # pragma: no cover
     pass
 
-__all__ = ['_NUMEXPR_INSTALLED']
+__all__ = ['_NUMEXPR_INSTALLED', '_NUMEXPR_VERSION']