diff options
-rwxr-xr-x | python/run-tests.py | 3 |
1 files changed, 2 insertions, 1 deletions
diff --git a/python/run-tests.py b/python/run-tests.py index f5857f8c62..ee73eb1506 100755 --- a/python/run-tests.py +++ b/python/run-tests.py @@ -56,7 +56,8 @@ LOGGER = logging.getLogger() def run_individual_python_test(test_name, pyspark_python): env = dict(os.environ) - env.update({'SPARK_TESTING': '1', 'PYSPARK_PYTHON': which(pyspark_python)}) + env.update({'SPARK_TESTING': '1', 'PYSPARK_PYTHON': which(pyspark_python), + 'PYSPARK_DRIVER_PYTHON': which(pyspark_python)}) LOGGER.debug("Starting test(%s): %s", pyspark_python, test_name) start_time = time.time() try: |