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