check if module existed

mdrachuk · Aug 22, 2019 · d8b50c1 · d8b50c1
1 parent a62471d
commit d8b50c1
Showing 1 changed file with 3 additions and 3 deletions.
diff --git a/cv.py b/cv.py
@@ -79,10 +79,10 @@ def _parse_args(args: List[str]):
 def _resolve_module(module_name: str):
     """Black magic. Prevents loading a package from cv dependencies."""
     invalidate_caches()
-    old_module = sys.modules.get(module_name, None)
-    del sys.modules[module_name]
+    old_module = sys.modules.pop(module_name, None)
     module = import_module(module_name)
-    sys.modules[module_name] = old_module
+    if old_module:
+        sys.modules[module_name] = old_module
     return module