Slim down main env

.gitattributes

@@ -6,3 +6,5 @@
 *.bat text eol=crlf
 
 pixi.lock linguist-language=YAML linguist-generated=true
+# GitHub syntax highlighting
+pixi.lock linguist-language=YAML

.github/workflows/array-api.yml

@@ -14,7 +14,7 @@ jobs:
   array-api-tests:
     name: Array API test
     timeout-minutes: 90
-    runs-on: ubuntu-latest-16core
+    runs-on: ubuntu-latest-8core
     steps:
       - name: Checkout branch
         uses: actions/checkout@v4

.github/workflows/ci.yml

@@ -5,7 +5,6 @@ on:
       - main
   pull_request:
 
-# Automatically stop old builds on the same branch/PR
 concurrency:
   group: ${{ github.workflow }}-${{ github.ref }}
   cancel-in-progress: true

.gitignore

@@ -406,3 +406,6 @@ output/
 # Type-generated
 ndonnx/_types.pyi
 ndonnx/__init__.pyi
+# pixi environments
+.pixi
+*.egg-info

.pre-commit-config.yaml

@@ -56,12 +56,6 @@ repos:
         language: system
         types: [text]
         files: \.(md|yml|yaml)$
-      # taplo
-      - id: taplo
-        name: taplo
-        entry: pixi run -e lint taplo format
-        language: system
-        types: [toml]
       # pre-commit-hooks
       - id: trailing-whitespace-fixer
         name: trailing-whitespace-fixer
@@ -73,11 +67,6 @@ repos:
         entry: pixi run -e lint end-of-file-fixer
         language: system
         types: [text]
-      - id: check-merge-conflict
-        name: check-merge-conflict
-        entry: pixi run -e lint check-merge-conflict --assume-in-merge
-        language: system
-        types: [text]
       # typos
       - id: typos
         name: typos

README.md

@@ -26,11 +26,15 @@ You can install the package in development mode using:
 ```bash
 git clone https://github.com/quantco/ndonnx
 cd ndonnx
+
+# For Array API tests
 git submodule update --init --recursive
 
-pixi run pre-commit-install
-pixi run postinstall
-pixi run test
+# Set up development environment
+pixi shell
+pre-commit install
+pip install -e .
+pytest tests -n auto
 ```
 
 ## Quick start
@@ -100,5 +104,5 @@ Summary(1119 total):
 Run the tests with:
 
 ```bash
-ARRAY_API_TESTS_MODULE=ndonnx pytest array_api_tests/ --json-report --json-report-file=api-coverage-tests.json -n auto
+pixi run arrayapitests
 ```

pixi.toml

@@ -5,14 +5,11 @@ platforms = ["osx-arm64", "osx-64", "linux-64", "linux-aarch64", "win-64"]
 
 [tasks]
 postinstall = "pip install --no-build-isolation --no-deps --disable-pip-version-check -e ."
+arrayapitests = "ARRAY_API_TESTS_MODULE=ndonnx pytest --max-examples 2 api-coverage-tests/array_api_tests/ --json-report --json-report-file=api-coverage-tests.json -n auto -vv --skips-file=skips.txt"
 
 [dependencies]
 python = ">=3.10"
-ndindex = "*"
 numpy = "*"
-onnx = ">=1.15"
-onnxruntime = "*"
-pandas = "*"
 spox = ">=0.10"
 
 [host-dependencies]
@@ -41,9 +38,9 @@ pytest-cov = "*"
 pytest-json-report = "*"
 pytest-xdist = "*"
 hypothesis = "*"
-mypy = ">=1.9,<1.10"
-pandas-stubs = "*"
 jupyterlab = "*"
+onnxruntime = ">=1.18.0,<1.19"
+onnx = ">=1.16.1,<1.17"
 [feature.test.tasks]
 test = "pytest"
 test-coverage = "pytest --cov=ndonnx --cov-report=xml --cov-report=term-missing"
@@ -54,7 +51,6 @@ insert-license-header = "*"
 docformatter = "*"
 ruff = "*"
 prettier = "*"
-taplo = "*"
 pre-commit-hooks = "*"
 typos = "*"
 [feature.lint.tasks]

pyproject.toml

@@ -80,3 +80,12 @@ exclude = ["docs/"]
 
 [tool.typos.default]
 extend-ignore-identifiers-re = ["scatter_nd", "arange"]
+
+[tool.pixi.project]
+channels = ["conda-forge"]
+platforms = ["osx-arm64"]
+
+[tool.pixi.pypi-dependencies]
+ndonnx = { path = ".", editable = true }
+
+[tool.pixi.tasks]