Merge pull request #3793 from AI-Hypercomputer:skip_hlo_tests

PiperOrigin-RevId: 908900427

Author: Google-ML-Automation

.github/workflows/promote_docker_image.yml

@@ -73,6 +73,7 @@ jobs:
       base_image: ${{ inputs.image_name }}:${{ inputs.image_tag }}
       is_scheduled_run: true
       maxtext_installed: true
+      additional_pytest_args: ${{ contains(inputs.image_name, 'maxtext_jax_nightly') && '--ignore=tests/integration/hlo_diff_test.py' || '' }}
 
   tagging:
     needs: [test]

.github/workflows/run_tests_coordinator.yml

@@ -51,6 +51,11 @@ on:
         required: false
         type: boolean
         default: false
+      additional_pytest_args:
+        description: 'Additional pytest arguments to pass during test execution'
+        required: false
+        type: string
+        default: ''
 
 permissions:
   contents: read
@@ -137,6 +142,7 @@ jobs:
             "cpu-unit": "--ignore=tests/post_training",
             "cpu-post-training-unit": ""
           }')[inputs.flavor] }}
+        ${{ inputs.additional_pytest_args }}
 
       # Resource Scaling
       xla_python_client_mem_fraction: "${{ contains(inputs.flavor, 'gpu') && '0.65' || '0.75' }}"