diff options
Diffstat (limited to 'pkgs/development/python-modules/jmp/default.nix')
-rw-r--r-- | pkgs/development/python-modules/jmp/default.nix | 23 |
1 files changed, 10 insertions, 13 deletions
diff --git a/pkgs/development/python-modules/jmp/default.nix b/pkgs/development/python-modules/jmp/default.nix index 945e99310366a..b0ec259f0723c 100644 --- a/pkgs/development/python-modules/jmp/default.nix +++ b/pkgs/development/python-modules/jmp/default.nix @@ -1,9 +1,10 @@ -{ buildPythonPackage -, fetchFromGitHub -, jax -, jaxlib -, lib -, pytestCheckHook +{ + buildPythonPackage, + fetchFromGitHub, + jax, + jaxlib, + lib, + pytestCheckHook, }: buildPythonPackage rec { @@ -19,13 +20,9 @@ buildPythonPackage rec { }; # Wheel requires only `numpy`, but the import needs `jax`. - propagatedBuildInputs = [ - jax - ]; + propagatedBuildInputs = [ jax ]; - pythonImportsCheck = [ - "jmp" - ]; + pythonImportsCheck = [ "jmp" ]; nativeCheckInputs = [ jaxlib @@ -33,7 +30,7 @@ buildPythonPackage rec { ]; meta = with lib; { - description = "This library implements support for mixed precision training in JAX."; + description = "This library implements support for mixed precision training in JAX"; homepage = "https://github.com/deepmind/jmp"; license = licenses.asl20; maintainers = with maintainers; [ ndl ]; |