From fc6b149c8c27cbec1909b9dd01f766cce68da6ac Mon Sep 17 00:00:00 2001 From: Anthony Mahanna Date: Sat, 23 Dec 2023 16:36:11 -0500 Subject: [PATCH] Update benchmark.yml --- .github/workflows/benchmark.yml | 9 +++------ 1 file changed, 3 insertions(+), 6 deletions(-) diff --git a/.github/workflows/benchmark.yml b/.github/workflows/benchmark.yml index a4dd1fe..aaa4aa9 100644 --- a/.github/workflows/benchmark.yml +++ b/.github/workflows/benchmark.yml @@ -101,10 +101,6 @@ jobs: pip install torch-scatter torch-sparse -f https://data.pyg.org/whl/torch-$(python -c 'import torch; print(torch.__version__.split("+")[0])')+cpu.html pip install -e '.[dev, tracing]' - - name: Declare some variables - shell: bash - run: echo "sha_short=$(git rev-parse --short "$GITHUB_SHA")" >> "$GITHUB_ENV" - - name: Write PR traces run: python benchmark/write.py --output_dir branch @@ -121,7 +117,8 @@ jobs: - name: Prepare PR Message run: | echo 'MESSAGE<> $GITHUB_ENV - echo "Benchmark ${sha_short}" >> $GITHUB_ENV + echo "Benchmark (${github.event.pull_request.head.sha})" >> $GITHUB_ENV + echo '' >> $GITHUB_ENV echo 'Root Span Comparisons:' >> $GITHUB_ENV echo '```json' >> $GITHUB_ENV echo "$ROOT_SPAN_COMPARISON" | jq . >> $GITHUB_ENV @@ -155,7 +152,7 @@ jobs: # echo "EOF" >> $GITHUB_ENV - name: Comment on PR - uses: actions/github-script@v5 + uses: actions/github-script@v7 with: github-token: ${{secrets.GITHUB_TOKEN}} script: |