Fix release wheels build.

PiperOrigin-RevId: 788247912

Author: tensorflower-gardener

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"