From d8b50c103b8d379017e7dc5e2338ddf096c275cc Mon Sep 17 00:00:00 2001 From: mdrachuk Date: Thu, 22 Aug 2019 17:30:50 +0300 Subject: [PATCH] check if module existed --- cv.py | 6 +++--- 1 file changed, 3 insertions(+), 3 deletions(-) diff --git a/cv.py b/cv.py index d1ef908..9186ec0 100755 --- 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