fix op benchmark ci error caused by missing test_pr branch, test=document_fix (#31920)

4 years ago · 17030ff28b
parent a71d72d921
commit 17030ff28b
1 changed files with 1 additions and 1 deletions
--- a/tools/test_op_benchmark.sh
+++ b/tools/test_op_benchmark.sh
@ -187,7 +187,7 @@ function run_op_benchmark_test {
  done
  # install tensorflow for testing accuary
  pip install tensorflow==2.3.0 tensorflow-probability
-  for branch_name in "develop" "test_pr"
+  for branch_name in "develop" "test"
  do
    git checkout $branch_name
    [ $? -ne 0 ] && LOG "[FATAL] Missing branch ${branch_name}." && exit 7