Something went wrong. Try again.
What happens if I push all of nixpkgs
Something went wrong. Try again.
1.2 kB · 63 lines
Nix
12345678910111213141516171819202122232425262728293031323334353637383940414243444546474849505152535455565758596061626364{ lib, buildPythonPackage, fetchFromGitHub, pytestCheckHook, setuptools, setuptools-scm, numpy, jaxlib, jax, torch, dask, sparse, array-api-strict, config, cudaSupport ? config.cudaSupport, cupy,}:
buildPythonPackage rec { pname = "array-api-compat"; version = "1.12"; pyproject = true;
src = fetchFromGitHub { owner = "data-apis"; repo = "array-api-compat"; tag = version; hash = "sha256-Hb0bFjVMl4CBI3gN3abTO2QUPAOvUaFE0GdPjdops5E="; };
build-system = [ setuptools setuptools-scm ];
nativeCheckInputs = [ pytestCheckHook numpy jaxlib jax torch dask sparse array-api-strict ] ++ lib.optionals cudaSupport [ cupy ];
pythonImportsCheck = [ "array_api_compat" ];
# CUDA (used via cupy) is not available in the testing sandbox disabledTests = [ "cupy" ];
meta = { homepage = "https://data-apis.org/array-api-compat"; changelog = "https://github.com/data-apis/array-api-compat/releases/tag/${src.tag}"; description = "Compatibility layer for NumPy to support the Python array API"; license = lib.licenses.mit; maintainers = with lib.maintainers; [ berquist ]; };}