diff --git a/.github/workflows/_sandbox.yaml b/.github/workflows/_sandbox.yaml index 3b663627b..ca593aa89 100644 --- a/.github/workflows/_sandbox.yaml +++ b/.github/workflows/_sandbox.yaml @@ -7,5 +7,8 @@ on: jobs: ngc-release: uses: ./.github/workflows/ngc-release-testing.yaml + with: + JAX_IMAGE: nvcr.io/nvidian/trigger-jax-toolbox-actions:14090586-jax-py3 + PAX_IMAGE: nvcr.io/nvidian/trigger-jax-toolbox-actions:14090586-pax-py3 secrets: inherit diff --git a/.github/workflows/ngc-release-testing.yaml b/.github/workflows/ngc-release-testing.yaml index 016209f31..3a678bc5e 100644 --- a/.github/workflows/ngc-release-testing.yaml +++ b/.github/workflows/ngc-release-testing.yaml @@ -9,12 +9,12 @@ on: type: string description: "JAX image to run tests on" required: false - default: '' + default: 'nvcr.io/nvidian/trigger-jax-toolbox-actions:14090586-jax-py3' PAX_IMAGE: type: string description: "Rosetta Pax image to run tests on" required: false - default: '' + default: 'nvcr.io/nvidian/trigger-jax-toolbox-actions:14090586-pax-py3' concurrency: group: ${{ github.workflow }}-${{ github.head_ref || github.run_id }}