From 0c2c80adc2372411c5fb2e7ac13d85a3d38c1df4 Mon Sep 17 00:00:00 2001
From: Adrian Seyboldt <adrian.seyboldt@gmail.com>
Date: Thu, 2 May 2024 17:38:55 +0200
Subject: [PATCH] ci: Fix architectures in CI

---
 .github/workflows/ci.yml | 17 +++++++++++++----
 1 file changed, 13 insertions(+), 4 deletions(-)

diff --git a/.github/workflows/ci.yml b/.github/workflows/ci.yml
index 0abf935..22fc6ef 100644
--- a/.github/workflows/ci.yml
+++ b/.github/workflows/ci.yml
@@ -26,7 +26,7 @@ jobs:
       matrix:
         # If you update the targets here, be sure to also update the
         # download-artifact step in the release job!
-        target: [x86_64]
+        target: [x86_64, aarch64, riscv64]
         # pytensor is currently giving me issues on the other targets...
         #target: [x86_64, aarch64, s390x, ppc64le]
     steps:
@@ -58,7 +58,7 @@ jobs:
           pip install pytest
           pytest
       - name: pytest
-        if: ${{ !startsWith(matrix.target, 'x86') && matrix.target != 'ppc64' }}
+        if: ${{ !startsWith(matrix.target, 'x86') }}
         uses: uraimo/run-on-arch-action@v2.7.2
         with:
           arch: ${{ matrix.target }}
@@ -70,7 +70,8 @@ jobs:
             pip3 install -U pip pytest
           run: |
             set -e
-            pip3 install 'nutpie[all]' --find-links dist --force-reinstall
+            pip3 install --find-links dist --force-reinstall 'nutpie[all]'
+            pip3 install 'nutpie[all]' --find-links dist --no-deps --no-index --force-reinstall
             pytest
 
   windows:
@@ -147,7 +148,7 @@ jobs:
           name: macos-${{ matrix.target }}-wheels
           path: dist
       - name: pytest
-        if: ${{ !startsWith(matrix.target, 'aarch64') }}
+        if: ${{ startsWith(matrix.target, 'aarch64') }}
         shell: bash
         run: |
           set -e
@@ -183,6 +184,14 @@ jobs:
         with:
           name: linux-x86_64-wheels
           path: dist
+      - uses: actions/download-artifact@v4
+        with:
+          name: linux-aarch64-wheels
+          path: dist
+      - uses: actions/download-artifact@v4
+        with:
+          name: linux-riscv64-wheels
+          path: dist
       - uses: actions/download-artifact@v4
         with:
           name: windows-x64-wheels