nordic-dev.net/nixpkgs
nordic-dev.net/nixpkgs
2
0
Fork 0
mirror of https://github.com/NixOS/nixpkgs.git synced 2025-01-04 03:53:56 +00:00
Code Issues Packages Projects Releases Wiki Activity

python312Packages.dm-haiku: 0.0.12 -> 0.0.13

Browse Source
This commit is contained in:
R. Ryantm 2024-11-30 08:47:33 +00:00 committed by natsukium
parent e383460f09
commit 379aeabd48
No known key found for this signature in database
GPG Key ID: 9EA45A31DB994C53
1 changed files with 17 additions and 6 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

23
pkgs/development/python-modules/dm-haiku/default.nix
View File

@ -3,8 +3,10 @@
buildPythonPackage,
fetchFromGitHub,
fetchpatch,
setuptools,
absl-py,
flax,
jax,
jaxlib,
jmp,
numpy,
@ -25,14 +27,14 @@
let
dm-haiku = buildPythonPackage rec {
pname = "dm-haiku";
version = "0.0.12";
format = "setuptools";
version = "0.0.13";
pyproject = true;
src = fetchFromGitHub {
owner = "deepmind";
repo = "dm-haiku";
rev = "refs/tags/v${version}";
hash = "sha256-aJRXlMq4CNMH3ZSTDP8MgnVltdSc8l5raw4//KccL48=";
hash = "sha256-RJpQ9BzlbQ4X31XoJFnsZASiaC9fP2AdyuTAGINhMxs=";
};
patches = [
@ -44,15 +46,24 @@ let
})
];
propagatedBuildInputs = [
build-system = [ setuptools ];
dependencies = [
absl-py
flax
jaxlib
jaxlib # implicit runtime dependency
jmp
numpy
tabulate
];
optional-dependencies = {
jax = [
jax
jaxlib
];
flax = [ flax ];
};
pythonImportsCheck = [ "haiku" ];
nativeCheckInputs = [