Refactor GitHub Actions workflow for testing

.github/workflows/jax-array-api.yml

@@ -7,15 +7,32 @@ on:
   pull_request:
     branches:
       - main
+  workflow_dispatch:
+    inputs:
+      halt-for-connection:
+        description: 'Should this workflow run wait for a remote connection?'
+        type: choice
+        required: true
+        default: 'yes'
+        options:
+          - 'yes'
+          - 'no'
 
 concurrency:
   group: ${{ github.workflow }}-${{ github.head_ref || github.ref }}
   cancel-in-progress: true
 permissions: {}
 jobs:
   build:
-    runs-on: linux-x86-n4-16
+    runs-on: linux-x86-g2-48-l4-4gpu
     container: us-docker.pkg.dev/ml-oss-artifacts-published/ml-public-container/ml-build:latest
+    # Service containers to run with `container-job`
+    services:
+      # Label used to access the service container
+      redis:
+        # Docker Hub image
+        image: redis
+
     strategy:
       matrix:
         python-version: [3.11]
@@ -26,6 +43,8 @@ jobs:
       uses: actions/checkout@08c6903cd8c0fde910a37f88322edcfb5dd907a8  # v5.0.0
       with:
         persist-credentials: false
+    - name: Wait For Connection
+      run: sleep 100000
     - name: Checkout array-api-tests
       uses: actions/checkout@08c6903cd8c0fde910a37f88322edcfb5dd907a8  # v5.0.0
       with:

.github/workflows/test_workflow

@@ -0,0 +1,22 @@
+name: Testing
+on:
+  pull_request:
+    branches:
+      - main
+  workflow_dispatch: # allows triggering the workflow run manually
+  workflow_call:
+
+permissions: {}
+
+jobs:
+  test:
+    runs-on: linux-x86-g2-16-l4-1gpu-quoct
+    container: 'us-docker.pkg.dev/ml-oss-artifacts-published/ml-public-container/ml-build-arm64:latest'
+    steps:
+      - name: Checkout code
+        uses: actions/checkout@v4
+      - name: Simulate build
+        run: |
+          echo "Building the project..."
+          sleep 100
+          echo "Build complete."