nordic-dev.net/nixpkgs
nordic-dev.net/nixpkgs
2
0
Fork 0
mirror of https://github.com/NixOS/nixpkgs.git synced 2025-01-22 12:53:54 +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, buildPythonPackage,
fetchFromGitHub, fetchFromGitHub,
fetchpatch, fetchpatch,
setuptools,
absl-py, absl-py,
flax, flax,
jax,
jaxlib, jaxlib,
jmp, jmp,
numpy, numpy,
@ -25,14 +27,14 @@
let let
dm-haiku = buildPythonPackage rec { dm-haiku = buildPythonPackage rec {
pname = "dm-haiku"; pname = "dm-haiku";
version = "0.0.12"; version = "0.0.13";
format = "setuptools"; pyproject = true;
src = fetchFromGitHub { src = fetchFromGitHub {
owner = "deepmind"; owner = "deepmind";
repo = "dm-haiku"; repo = "dm-haiku";
rev = "refs/tags/v${version}"; rev = "refs/tags/v${version}";
hash = "sha256-aJRXlMq4CNMH3ZSTDP8MgnVltdSc8l5raw4//KccL48="; hash = "sha256-RJpQ9BzlbQ4X31XoJFnsZASiaC9fP2AdyuTAGINhMxs=";
}; };
patches = [ patches = [
@ -44,15 +46,24 @@ let
}) })
]; ];
propagatedBuildInputs = [ build-system = [ setuptools ];
dependencies = [
absl-py absl-py
flax jaxlib # implicit runtime dependency
jaxlib
jmp jmp
numpy numpy
tabulate tabulate
]; ];
optional-dependencies = {
jax = [
jax
jaxlib
];
flax = [ flax ];
};
pythonImportsCheck = [ "haiku" ]; pythonImportsCheck = [ "haiku" ];
nativeCheckInputs = [ nativeCheckInputs = [