Skip to content

Commit 0f63bea

Browse files
quoctruongGoogle-ML-Automation
authored andcommitted
Test hermetic C++ container
PiperOrigin-RevId: 799690365
1 parent 55b5aef commit 0f63bea

File tree

1 file changed

+4
-1
lines changed

1 file changed

+4
-1
lines changed

.github/workflows/wheel_tests_nightly_release.yml

Lines changed: 4 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -18,6 +18,9 @@
1818
name: CI - Wheel Tests (Nightly/Release)
1919

2020
on:
21+
pull_request:
22+
branches:
23+
- main
2124
workflow_dispatch:
2225
inputs:
2326
gcs_download_uri:
@@ -272,7 +275,7 @@ jobs:
272275
matrix:
273276
python: ["3.11", "3.13", "3.13-nogil"]
274277
cuda-version: [12, 13]
275-
container: "us-docker.pkg.dev/ml-oss-artifacts-published/ml-public-container/ml-build:latest"
278+
container: "us-docker.pkg.dev/ml-oss-artifacts-transient/ml-public-testing-container/ml-build:infrastructure-public-image-a9244380692c"
276279

277280
# Verifies that JAX's release wheels can be installed
278281
name: "Verify release wheels install (Python ${{ matrix.python }})"

0 commit comments

Comments
 (0)