Refactor matrix

bigscience-workshop · Aug 8, 2023 · dde6a1a · dde6a1a
1 parent 015238a
commit dde6a1a
Showing 1 changed file with 4 additions and 7 deletions.
diff --git a/.github/workflows/run-tests.yaml b/.github/workflows/run-tests.yaml
@@ -10,12 +10,9 @@ jobs:
     runs-on: ubuntu-latest
     strategy:
       matrix:
-        include:
-          - {python-version: '3.8', model: 'bigscience/bloom-560m', adapter: 'artek0chumak/bloom-560m-safe-peft'}
-          - {python-version: '3.9', model: 'bigscience/bloom-560m', adapter: 'artek0chumak/bloom-560m-safe-peft'}
-          - {python-version: '3.10', model: 'bigscience/bloom-560m', adapter: 'artek0chumak/bloom-560m-safe-peft'}
-          - {python-version: '3.11', model: 'bigscience/bloom-560m', adapter: 'artek0chumak/bloom-560m-safe-peft'}
-      fail-fast: true
+        model: [ 'bigscience/bloom-560m', 'Maykeye/TinyLLama-v0' ]
+        python-version: [ '3.8', '3.9', '3.10', '3.11' ]
+      fail-fast: false
     timeout-minutes: 15
     steps:
       - name: Checkout
@@ -37,7 +34,7 @@ jobs:
         run: |
           export MODEL_NAME=${{ matrix.model }}
           export REF_NAME=${{ matrix.model }}
-          export ADAPTER_NAME=${{ matrix.adapter }}
+          export ADAPTER_NAME=${{ matrix.model == 'bigscience/bloom-560m' && 'artek0chumak/bloom-560m-safe-peft' || '' }}
 
           python -m petals.cli.run_dht --identity tests/test.id --host_maddrs /ip4/127.0.0.1/tcp/31337 &> bootstrap.log &
           BOOTSTRAP_PID=$!