diff --git a/.github/workflows/model_profiling.yml b/.github/workflows/model_profiling.yml index d53257d4c..1b22fdc66 100644 --- a/.github/workflows/model_profiling.yml +++ b/.github/workflows/model_profiling.yml @@ -82,10 +82,7 @@ jobs: PROFILE_MODE: ${{ github.event_name == 'pull_request' && 'summary' || github.event.inputs.profile_mode || 'trace' }} POLICY_FILTER: ${{ github.event_name == 'pull_request' && 'act' || github.event.inputs.policies || '' }} RESULTS_REPO: ${{ github.event.inputs.results_repo || 'model-profiling-history' }} - SHOULD_PUBLISH: ${{ github.event_name == 'pull_request' || github.event_name == 'schedule' || (github.event_name == 'workflow_dispatch' && github.event.inputs.publish_results == 'true') }} - GIT_REF: ${{ github.head_ref || github.ref_name || github.event.inputs.git_ref || 'main' }} - PR_NUMBER: ${{ github.event.pull_request.number || '' }} - HOST_GIT_COMMIT: ${{ github.event.pull_request.head.sha || github.event.inputs.git_commit || github.sha }} + SHOULD_PUBLISH: ${{ github.event_name == 'schedule' || (github.event_name == 'workflow_dispatch' && github.event.inputs.publish_results == 'true') }} steps: - uses: actions/checkout@de0fac2e4500dabe0009e67214ff5f5447ce83dd # v6.0.2 with: @@ -97,6 +94,10 @@ jobs: run: docker pull huggingface/lerobot-gpu:latest - name: Run model profiling + env: + HOST_GIT_COMMIT: ${{ github.event.pull_request.head.sha || github.event.inputs.git_commit || github.sha }} + PROFILE_GIT_REF: ${{ github.head_ref || github.ref_name || github.event.inputs.git_ref || 'main' }} + PROFILE_PR_NUMBER: ${{ github.event.pull_request.number || '' }} run: | set -eux mkdir -p profiling-results @@ -114,8 +115,8 @@ jobs: -e XDG_DATA_HOME=/tmp/xdg-data \ -e XDG_CACHE_HOME=/tmp/xdg-cache \ -e HOST_GIT_COMMIT="${HOST_GIT_COMMIT}" \ - -e GIT_REF="${GIT_REF}" \ - -e PR_NUMBER="${PR_NUMBER}" \ + -e PROFILE_GIT_REF="${PROFILE_GIT_REF}" \ + -e PROFILE_PR_NUMBER="${PROFILE_PR_NUMBER}" \ -e HF_USER_TOKEN="${HF_USER_TOKEN}" \ -e HF_TOKEN="${HF_USER_TOKEN}" \ -e PROFILE_MODE="${PROFILE_MODE}" \ @@ -200,8 +201,8 @@ jobs: --results_repo="${RESULTS_REPO}" --profile_mode="${PROFILE_MODE}" --git_commit="${HOST_GIT_COMMIT}" - --git_ref="${GIT_REF}" - --pr_number="${PR_NUMBER}" + --git_ref="${PROFILE_GIT_REF}" + --pr_number="${PROFILE_PR_NUMBER}" ) if [[ -n "${POLICY_FILTER}" ]]; then