build: pixi environment setup (#780)

Author: hameerabbasi

.gitattributes

@@ -1 +1,3 @@
 sparse/_version.py export-subst
+# GitHub syntax highlighting
+pixi.lock linguist-language=YAML linguist-generated=true

.gitignore

@@ -82,3 +82,7 @@ results/
 
 # Notebooks converted to scripts.
 docs/examples_ipynb/
+
+# Pixi envs
+.pixi/
+pixi.lock

ci/environment.yml

@@ -4,6 +4,7 @@ channels:
   - nodefaults
 dependencies:
   - python
+  - pip
   - numpy
   - numba
   - scipy

pixi.toml

@@ -0,0 +1,66 @@
+[project]
+authors = ["Hameer Abbasi <[email protected]>"]
+channels = ["conda-forge"]
+name = "sparse"
+platforms = ["osx-arm64", "osx-64", "linux-64", "win-64"]
+
+[pypi-dependencies]
+sparse = { path = ".", editable = true }
+
+[dependencies]
+python = ">=3.10"
+numpy = ">=1.17"
+numba = ">=0.49"
+
+[feature.extras.dependencies]
+dask = ">=2024"
+scipy = ">=0.19"
+scikit-learn = "*"
+
+[feature.docs.pypi-dependencies]
+mkdocs-material = "*"
+mkdocstrings = { version = "*", extras = ["python"] }
+mkdocs-gen-files = "*"
+mkdocs-literate-nav = "*"
+mkdocs-section-index = "*"
+mkdocs-jupyter = "*"
+
+[feature.tests.tasks]
+test = "pytest --pyargs sparse -n auto"
+test-mlir = { cmd = "pytest --pyargs sparse/mlir_backend -vvv", env = { SPARSE_BACKEND = "MLIR" } }
+test-finch = { cmd = "pytest --pyargs sparse/tests  -n auto", env = { SPARSE_BACKEND = "Finch" }, depends-on = ["precompile"] }
+
+[feature.tests.dependencies]
+pytest = ">=3.5"
+pytest-cov = "*"
+pytest-xdist = "*"
+pre-commit = "*"
+pytest-codspeed = "*"
+
+[feature.notebooks.dependencies]
+nbmake = "*"
+matplotlib = "*"
+
+[feature.matrepr.dependencies]
+matrepr = "*"
+
+[feature.finch.tasks]
+precompile = "python -c 'import finch'"
+
+[feature.finch.pypi-dependencies]
+scipy = ">=0.19"
+finch-tensor = ">=0.1.31"
+
+[feature.mlir]
+channel-priority = "disabled"
+channels = ["conda-forge", "conda-forge/label/llvm_rc"]
+
+[feature.mlir.dependencies]
+scipy = ">=0.19"
+mlir-python-bindings = "19.*"
+
+[environments]
+tests = ["tests", "extras"]
+docs = ["docs", "extras"]
+mlir-dev = ["tests", "mlir"]
+finch-dev = ["tests", "finch"]

pyproject.toml

@@ -29,7 +29,6 @@ classifiers = [
 [project.optional-dependencies]
 docs = [
         "mkdocs-material",
-        "mkdocstrings",
         "mkdocstrings[python]",
         "mkdocs-gen-files",
         "mkdocs-literate-nav",