1{
2 lib,
3 buildPythonPackage,
4 fetchFromGitHub,
5 numpy,
6 setuptools,
7}:
8
9buildPythonPackage rec {
10 pname = "jumpy";
11 version = "1.0.0";
12
13 src = fetchFromGitHub {
14 owner = "Farama-Foundation";
15 repo = "jumpy";
16 rev = version;
17 hash = "sha256-tPQ/v2AVnAEC+08BVAvvgJ8Pj89nXZSn2tQ6nxXuSfA=";
18 };
19
20 format = "pyproject";
21
22 nativeBuildInputs = [ setuptools ];
23
24 propagatedBuildInputs = [ numpy ];
25
26 pythonImportsCheck = [ "jumpy" ];
27
28 meta = with lib; {
29 description = "Jumpy is a common backend for NumPy and optionally JAX";
30 homepage = "https://github.com/Farama-Foundation/Jumpy";
31 license = licenses.asl20;
32 maintainers = with maintainers; [ GaetanLepage ];
33 };
34}