refactor JAX unit test reporting

NVIDIA · Sep 1, 2023 · 118e282 · 118e282
1 parent 1b822fa
commit 118e282
Showing 1 changed file with 6 additions and 6 deletions.
diff --git a/.github/workflows/_test_jax.yaml b/.github/workflows/_test_jax.yaml
@@ -39,8 +39,8 @@ jobs:
       - name: Print environment variables
         run: env
 
-      - name: Print GPU information
-        run: nvidia-smi  
+      # - name: Print GPU information
+      #   run: nvidia-smi  
 
       - name: Login to GitHub Container Registry
         uses: docker/login-action@v2
@@ -49,10 +49,10 @@ jobs:
           username: ${{ github.repository_owner }}
           password: ${{ secrets.GITHUB_TOKEN }}
 
-      - name: Pull JAX image
-        shell: bash -x -e {0}
-        run: |
-          docker pull ${{ inputs.JAX_IMAGE }}
+      # - name: Pull JAX image
+      #   shell: bash -x -e {0}
+      #   run: |
+      #     docker pull ${{ inputs.JAX_IMAGE }}
 
       # - name: Backend-independent tests
       #   shell: bash -x {0}