Merge pull request tensorflow#97728 from tensorflow/r2.20-b0b73e1d3b4

r2.20 cherry-pick: b0b73e1 "Fix release wheels build."

Author: learning-to-play

ci/official/envs/versions_upload

@@ -26,4 +26,4 @@ TFCI_ARTIFACT_STAGING_GCS_ENABLE=1
 TFCI_ARTIFACT_STAGING_GCS_URI="gs://tensorflow-ci-staging/staging/versions/${_LOUHI_EXECUTION_ID:-$(git rev-parse HEAD)}/"
 
 TFCI_BAZEL_COMMON_ARGS="$TFCI_BAZEL_COMMON_ARGS --config resultstore"
-TFCI_BUILD_PIP_PACKAGE_ARGS="$TFCI_BUILD_PIP_PACKAGE_ARGS --repo_env=ML_WHEEL_TYPE=release"
+TFCI_BUILD_PIP_PACKAGE_BASE_ARGS="$TFCI_BUILD_PIP_PACKAGE_BASE_ARGS --repo_env=ML_WHEEL_TYPE=release"