diff options
Diffstat (limited to 'pkgs/development/python-modules/treeo/default.nix')
-rw-r--r-- | pkgs/development/python-modules/treeo/default.nix | 29 |
1 files changed, 12 insertions, 17 deletions
diff --git a/pkgs/development/python-modules/treeo/default.nix b/pkgs/development/python-modules/treeo/default.nix index 789eb346e8c64..bca135f92a177 100644 --- a/pkgs/development/python-modules/treeo/default.nix +++ b/pkgs/development/python-modules/treeo/default.nix @@ -1,10 +1,11 @@ -{ buildPythonPackage -, fetchFromGitHub -, fetchpatch -, jax -, jaxlib -, lib -, poetry-core +{ + buildPythonPackage, + fetchFromGitHub, + fetchpatch, + jax, + jaxlib, + lib, + poetry-core, }: buildPythonPackage rec { @@ -34,22 +35,16 @@ buildPythonPackage rec { }) ]; - nativeBuildInputs = [ - poetry-core - ]; + nativeBuildInputs = [ poetry-core ]; # jax is not declared in the dependencies, but is necessary. - propagatedBuildInputs = [ - jax - ]; + propagatedBuildInputs = [ jax ]; nativeCheckInputs = [ jaxlib ]; - pythonImportsCheck = [ - "treeo" - ]; + pythonImportsCheck = [ "treeo" ]; meta = with lib; { - description = "A small library for creating and manipulating custom JAX Pytree classes."; + description = "Small library for creating and manipulating custom JAX Pytree classes"; homepage = "https://github.com/cgarciae/treeo"; license = licenses.mit; maintainers = with maintainers; [ ndl ]; |