Something went wrong. Try again.
What happens if I push all of nixpkgs
Something went wrong. Try again.
1.2 kB · 59 lines
Nix
at gcc-offload
123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960{ lib, buildPythonPackage, fetchFromGitHub, pytestCheckHook, setuptools, numpy, jaxlib, jax, torch, dask, sparse, array-api-strict, config, cudaSupport ? config.cudaSupport, cupy,}:
buildPythonPackage rec { pname = "array-api-compat"; version = "1.9.1"; pyproject = true;
src = fetchFromGitHub { owner = "data-apis"; repo = "array-api-compat"; rev = "refs/tags/${version}"; hash = "sha256-X6y6hX/HdkiLZkj9AOEYsZOlYhR7wUt9kQjHfMfWqIc="; };
build-system = [ setuptools ];
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 pytestFlagsArray = [ "-k" "'not cupy'" ];
meta = { homepage = "https://data-apis.org/array-api-compat"; changelog = "https://github.com/data-apis/array-api-compat/releases/tag/${version}"; description = "Compatibility layer for NumPy to support the Python array API"; license = lib.licenses.mit; maintainers = with lib.maintainers; [ berquist ]; };}