From f751b9b95c254468c158a6a40005510474e3fd1f Mon Sep 17 00:00:00 2001 From: Hyukjin Kwon Date: Fri, 5 Apr 2024 11:11:46 +0900 Subject: [PATCH] Make testing script to inherits SPARK_CONNECT_TESTING_REMOTE env --- python/run-tests.py | 3 +++ 1 file changed, 3 insertions(+) diff --git a/python/run-tests.py b/python/run-tests.py index 66514a9216e91..ebdd4a9a21798 100755 --- a/python/run-tests.py +++ b/python/run-tests.py @@ -98,6 +98,9 @@ def run_individual_python_test(target_dir, test_name, pyspark_python, keep_test_ 'PYARROW_IGNORE_TIMEZONE': '1', }) + if "SPARK_CONNECT_TESTING_REMOTE" in os.environ: + env.update({"SPARK_CONNECT_TESTING_REMOTE": os.environ["SPARK_CONNECT_TESTING_REMOTE"]}) + # Create a unique temp directory under 'target/' for each run. The TMPDIR variable is # recognized by the tempfile module to override the default system temp directory. tmp_dir = os.path.join(target_dir, str(uuid.uuid4()))