diff --git a/.github/workflows/gpu-tests.yml b/.github/workflows/gpu-tests.yml index 0aa02d4849..a2d6ca9a4a 100644 --- a/.github/workflows/gpu-tests.yml +++ b/.github/workflows/gpu-tests.yml @@ -39,7 +39,7 @@ jobs: python -m pip install --no-binary=protobuf protobuf - name: Get Branch name id: get-branch-name - uses: NVIDIA-Merlin/.github/actions/branch-name@main + uses: NVIDIA-Merlin/.github/actions/branch-name@branch-name-pull-request - name: Run tests run: | merlin_branch="${{ steps.get-branch-name.outputs.branch }}"