Something went wrong. Try again.
lol
Something went wrong. Try again.
1.2 kB · 68 lines
Nix
at master
123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869{ lib, config, buildPythonPackage, fetchFromGitHub,
# build-system setuptools, setuptools-scm,
# tests array-api-strict, dask, jax, jaxlib, numpy, pytestCheckHook, sparse, torch, cupy,
cudaSupport ? config.cudaSupport,}:
buildPythonPackage rec { pname = "array-api-compat"; version = "1.13"; pyproject = true;
src = fetchFromGitHub { owner = "data-apis"; repo = "array-api-compat"; tag = version; hash = "sha256-4ZoYtqkY/xPRyBP1xeMR80zMfwiaKtXX/3XzqiweCtc="; };
build-system = [ setuptools setuptools-scm ];
nativeCheckInputs = [ array-api-strict dask jax jaxlib numpy pytestCheckHook sparse torch ] ++ 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 ]; };}