Allow some dask computes

Author: crusaderky

pixi.lock

@@ -54,6 +54,7 @@ array-api-compat = ">=1.10.0,<2"
 array-api-extra = { path = ".", editable = true }
 
 [tool.pixi.feature.lint.dependencies]
+typing-extensions = "*"
 pre-commit = "*"
 pylint = "*"
 basedmypy = "*"
@@ -63,6 +64,9 @@ numpydoc = ">=1.8.0,<2"
 array-api-strict = "*"
 numpy = "*"
 pytest = "*"
+dask-core = "*"  # No distributed, tornado, etc.
+# NOTE: don't add jax, pytorch, sparse, cupy here
+# as they slow down mypy and are not portable across target OSs
 
 [tool.pixi.feature.lint.tasks]
 pre-commit-install = "pre-commit install"