Add default value

NVIDIA · Nov 14, 2023 · ee930f3 · ee930f3
1 parent cbba2e3
commit ee930f3
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/integration_tests/src/main/python/conftest.py b/integration_tests/src/main/python/conftest.py
@@ -78,7 +78,7 @@ def is_dataproc_runtime():
     return runtime_env() == "dataproc"
 
 def get_test_tz():
-    return os.environ["PYSP_TEST_spark_sql_session_timeZone"]
+    os.environ.get('PYSP_TEST_spark_sql_session_timeZone', 'UTC')
 
 def is_utc():
     return get_test_tz() == "UTC"