diff --git a/python/pyspark/daemon.py b/python/pyspark/daemon.py index bbbc495d053e..a23af109ea6d 100644 --- a/python/pyspark/daemon.py +++ b/python/pyspark/daemon.py @@ -30,7 +30,7 @@ from pyspark.serializers import read_int, write_int, write_with_length, UTF8Deserializer -if len(sys.argv) > 1: +if len(sys.argv) > 1 and sys.argv[1].startswith("pyspark"): import importlib worker_module = importlib.import_module(sys.argv[1])