about summary refs log tree commit diff
path: root/pkgs/development/python-modules/treeo/default.nix
diff options
context:
space:
mode:
Diffstat (limited to 'pkgs/development/python-modules/treeo/default.nix')
-rw-r--r--pkgs/development/python-modules/treeo/default.nix29
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 ];