Fix release script (pytorch#2736)

Summary:
Pull Request resolved: pytorch#2736

imported-using-ghimport

Test Plan: Imported from OSS

Reviewed By: Gasoonjia

Differential Revision: D55458620

Pulled By: guangy10

fbshipit-source-id: 240c02d4ca95792c9878d1f19d13abb2810dd53c

scripts/release/apply-release-changes.sh

@@ -11,9 +11,9 @@
 # Creates PR with release only changes.
 #
 # Usage (run from root of project):
-#  DRY_RUN=disabled ./scripts/release/apply-release-changes.sh
+#   TEST_INFRA_BRANCH=release/2.3 ./scripts/release/apply-release-changes.sh
 #
-# RELEASE_VERSION: Version of this current release
+# TEST_INFRA_BRANCH: The release branch of test-infra that houses all reusable
 '
 
 set -eou pipefail
@@ -37,11 +37,11 @@ fi
 echo "Applying release-only changes to workflows"
 for i in .github/workflows/*.yml; do
   if [[ "$OSTYPE" == "darwin"* ]]; then
-    sed -i '' -e s#@main#@"${RELEASE_BRANCH}"# $i;
-    sed -i '' -e s#test-infra-ref:[[:space:]]main#"test-infra-ref: ${RELEASE_BRANCH}"# $i;
+    sed -i '' -e s#@main#@"${TEST_INFRA_BRANCH}"# $i;
+    sed -i '' -e s#test-infra-ref:[[:space:]]main#"test-infra-ref: ${TEST_INFRA_BRANCH}"# $i;
   else
-    sed -i -e s#@main#@"${RELEASE_BRANCH}"# $i;
-    sed -i -e s#test-infra-ref:[[:space:]]main#"test-infra-ref: ${RELEASE_BRANCH}"# $i;
+    sed -i -e s#@main#@"${TEST_INFRA_BRANCH}"# $i;
+    sed -i -e s#test-infra-ref:[[:space:]]main#"test-infra-ref: ${TEST_INFRA_BRANCH}"# $i;
   fi
 done