From 1178bad3395228f34341b15eb58d6de4b68a7e66 Mon Sep 17 00:00:00 2001
From: Saran Tunyasuvunakool <stunya@google.com>
Date: Thu, 19 Oct 2023 11:12:00 -0700
Subject: [PATCH] Run MJX tests on GitHub Actions.

PiperOrigin-RevId: 574939328
Change-Id: Ic6e5650ef6a7df8c94a189206fddf06bd9c4eb97
---
 .github/workflows/build.yml | 23 ++++++++++++++++++++++-
 1 file changed, 22 insertions(+), 1 deletion(-)

diff --git a/.github/workflows/build.yml b/.github/workflows/build.yml
index 84646f60..7abd4c7e 100644
--- a/.github/workflows/build.yml
+++ b/.github/workflows/build.yml
@@ -219,7 +219,7 @@ jobs:
       working-directory: python/dist
       run: >
         source ${{ matrix.tmpdir }}/venv/bin/activate &&
-        pip install --no-deps --no-index mujoco-*.whl
+        pip install --no-index mujoco-*.whl
     - name: Test Python bindings
       if: ${{ runner.os != 'Windows' }}
       shell: bash
@@ -228,6 +228,27 @@ jobs:
       run: >
         source ${{ matrix.tmpdir }}/venv/bin/activate &&
         pytest -v --pyargs mujoco
+    - name: Package MJX
+      if: ${{ runner.os != 'Windows' }}
+      shell: bash
+      working-directory: mjx
+      run:
+        source ${{ matrix.tmpdir }}/venv/bin/activate &&
+        python -m build .
+    - name: Install MJX
+      if: ${{ runner.os != 'Windows' }}
+      shell: bash
+      working-directory: mjx
+      run:
+        source ${{ matrix.tmpdir }}/venv/bin/activate &&
+        pip install --require-hashes -r requirements.txt &&
+        pip install --no-index dist/mujoco_mjx-*.whl
+    - name: Test MJX
+      if: ${{ runner.os != 'Windows' }}
+      shell: bash
+      run:
+        source ${{ matrix.tmpdir }}/venv/bin/activate &&
+        pytest -n auto -v --pyargs mujoco.mjx
     - name: Notify team chat
       shell: bash
       env: