update env

Author: nitins17

ci/envs/run_tests/bazel_gpu_local

@@ -16,7 +16,7 @@
 source ci/envs/default
 
 # Enable local Bazel GPU tests
-export JAXCI_RUN_BAZEL_GPU_TEST_LOCAL=1
+export JAXCI_RUN_BAZEL_TEST_GPU_LOCAL=1
 
 # Only Linux x86 runs local GPU tests at the moment.
 export JAXCI_DOCKER_IMAGE="gcr.io/tensorflow-testing/nosla-cuda${JAXCI_DOCKER_CUDA_VERSION}-cudnn9.1-ubuntu20.04-manylinux2014-multipython"
@@ -28,4 +28,4 @@ export NCCL_DEBUG=WARN
 export JAX_ENABLE_X64=0
 
 # Clone XLA at HEAD.
-export JAXCI_CLONE_MAIN_XLA=1
+export JAXCI_CLONE_MAIN_XLA=1