Extract function for unloading pkg_resources. Ref #345.

1 files changed, 6 insertions, 1 deletions

diff --git a/ez_setup.py b/ez_setup.py
index 154cae0a..4bffc14c 100644
--- a/ez_setup.py
+++ b/ez_setup.py
@@ -175,10 +175,15 @@ def use_setuptools(
 
         # otherwise, unload pkg_resources to allow the downloaded version to
         #  take precedence.
-        del pkg_resources, sys.modules['pkg_resources']
+        del pkg_resources
+        _unload_pkg_resources()
         return _do_download(version, download_base, to_dir, download_delay)
 
 
+def _unload_pkg_resources():
+    del sys.modules['pkg_resources']
+
+
 def _clean_check(cmd, target):
     """
     Run the command to download target.