Allow specifying extra-requires in the GitHub Actions config

asmeurer · asmeurer · commit ae488ccd2fc8 · 2024-01-17T14:39:17.000-07:00
This is a change from data-apis#76.
diff --git a/.github/workflows/array-api-tests.yml b/.github/workflows/array-api-tests.yml
@@ -6,6 +6,12 @@ on:
       package-name:
         required: true
         type: string
+      module-name:
+        required: false
+        type: string
+      extra-requires:
+        required: false
+        type: string
       package-version:
         required: false
         type: string
@@ -54,12 +60,12 @@ jobs:
       if: "! (matrix.python-version == '3.11' && inputs.package-name == 'numpy' && contains(inputs.package-version, '1.21'))"
       run: |
         python -m pip install --upgrade pip
-        python -m pip install '${{ inputs.package-name }} ${{ inputs.package-version }}'
+        python -m pip install '${{ inputs.package-name }} ${{ inputs.package-version }}' ${{ inputs.extra-requires }}
         python -m pip install -r ${GITHUB_WORKSPACE}/array-api-tests/requirements.txt
     - name: Run the array API testsuite (${{ inputs.package-name }})
       if: "! (matrix.python-version == '3.11' && inputs.package-name == 'numpy' && contains(inputs.package-version, '1.21'))"
       env:
-        ARRAY_API_TESTS_MODULE: array_api_compat.${{ inputs.package-name }}
+        ARRAY_API_TESTS_MODULE: array_api_compat.${{ inputs.module-name || inputs.package-name }}
         # This enables the NEP 50 type promotion behavior (without it a lot of
         # tests fail on bad scalar type promotion behavior)
         NPY_PROMOTION_STATE: weak
