Fix mypy when prepending $PYTHONPATH to sys.path

Lightning-AI · Jan 28, 2021 · b52fa07 · b52fa07
1 parent 8cb2451
commit b52fa07
Showing 1 changed file with 4 additions and 2 deletions.
diff --git 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