Nightly build of MJX container + small perf test

NVIDIA · Jan 26, 2024 · 82f4b8f · 82f4b8f
1 parent 6fda38b
commit 82f4b8f
Showing 1 changed file with 7 additions and 6 deletions.
diff --git a/.github/workflows/nightly-mjx-test-unit.yaml b/.github/workflows/nightly-mjx-test-unit.yaml
@@ -2,16 +2,17 @@ name: Nightly MJX unit test
 run-name: Nightly MJX unit test (${{ github.event_name == 'workflow_run' && format('nightly {0}', github.event.workflow_run.created_at) || github.event_name }})
 
 on:
-  workflow_run:
-    workflows: [Nightly MJX build]
-    types: [completed]
-    branches: [main]
+  push:
+  # workflow_run:
+  #   workflows: [Nightly MJX build]
+  #   types: [completed]
+  #   branches: [main]
   workflow_dispatch:
     inputs:
       MJX_IMAGE:
         type: string
         description: 'MJX image built by NVIDIA/MJX-Toolbox'
-        default: 'ghcr.io/nvidia/upstream-mjx:latest'
+        default: ghcr.io/nvidia/jax-toolbox-internal:7662445215-upstream-mjx-amd64 #'ghcr.io/nvidia/upstream-mjx:latest'
         required: true
       PUBLISH:
         type: boolean
@@ -26,7 +27,7 @@ permissions:
   packages: write # to upload container
 
 env:
-  DEFAULT_MJX_IMAGE: ghcr.io/nvidia/upstream-mjx:latest
+  DEFAULT_MJX_IMAGE: ghcr.io/nvidia/jax-toolbox-internal:7662445215-upstream-mjx-amd64 #ghcr.io/nvidia/upstream-mjx:latest
   ARTIFACT_NAME: artifact-mjx-unit-test
 
 jobs: