Add ndindex back for Array API tests (#5)

Quantco · Jun 15, 2024 · 9cedf62 · 9cedf62
1 parent af5ad37
commit 9cedf62
Show file tree

Hide file tree

Showing 4 changed files with 39 additions and 1 deletion.
diff --git a/.github/workflows/array-api.yml b/.github/workflows/array-api.yml
@@ -19,7 +19,7 @@ jobs:
       github.event_name == 'schedule'
     name: Array API test
     timeout-minutes: 90
-    runs-on: ubuntu-latest-8core
+    runs-on: ubuntu-latest
     steps:
       - name: Checkout branch
         uses: actions/checkout@v4

diff --git a/README.md b/README.md
@@ -81,6 +81,7 @@ It has a couple of key features:
   ```python
   import onnxruntime as ort
   import numpy as np
+
   inference_session = ort.InferenceSession("mean_drop_outliers.onnx")
   prediction, = inference_session.run(None, {
       "x": np.array([-10, 0.5, 1, 5], dtype=np.float32),

diff --git a/pixi.lock b/pixi.lock
diff --git a/pixi.toml b/pixi.toml
@@ -42,6 +42,7 @@ hypothesis = "*"
 onnxruntime = "*"
 onnx = "*"
 mypy = ">=1.10.0,<1.11"
+ndindex = "*"
 [feature.test.tasks]
 test = "pytest"
 test-coverage = "pytest --cov=ndonnx --cov-report=xml --cov-report=term-missing"