From b52fa07d6e6cdf672f4563869d983d36cb653fe4 Mon Sep 17 00:00:00 2001 From: Sumanth Ratna Date: Thu, 28 Jan 2021 15:47:16 -0500 Subject: [PATCH] Fix mypy when prepending $PYTHONPATH to sys.path --- tests/models/data/horovod/train_default_model.py | 6 ++++-- 1 file changed, 4 insertions(+), 2 deletions(-) diff --git a/tests/models/data/horovod/train_default_model.py b/tests/models/data/horovod/train_default_model.py index f41bf59bb4f4c..23217cd829cbc 100644 --- a/tests/models/data/horovod/train_default_model.py +++ b/tests/models/data/horovod/train_default_model.py @@ -21,8 +21,10 @@ import os import sys -# this is need as e.g. Conda do not uses `PYTHONPATH` env var as pip or/and virtualenv -sys.path = os.getenv('PYTHONPATH').split(':') + sys.path +# this is needed because Conda does not use `PYTHONPATH` env var while pip and virtualenv do +PYTHONPATH = os.getenv('PYTHONPATH', '') +if ':' in PYTHONPATH: + sys.path = PYTHONPATH.split(':') + sys.path from pytorch_lightning import Trainer # noqa: E402 from pytorch_lightning.callbacks import ModelCheckpoint # noqa: E402