Check for None spec in module_util.py.

PiperOrigin-RevId: 303904448
Change-Id: Ib4a8cd50819aef30b1172b8b03aa461b48d88bca
This commit is contained in:
Anna R 2020-03-30 23:55:17 -07:00 committed by TensorFlower Gardener
parent 488ee20880
commit 71d8bea52c

View File

@ -59,7 +59,7 @@ def get_parent_dir_for_name(module_name):
spec = importlib.util.find_spec(name_split[0])
except ValueError:
return None
if not spec.origin:
if not spec or not spec.origin:
return None
base_path = os.path.dirname(spec.origin)
return os.path.join(base_path, *name_split[1:-1])