diff options
author | Samuel Ainsworth <skainsworth@gmail.com> | 2022-04-11 21:07:50 +0000 |
---|---|---|
committer | Samuel Ainsworth <skainsworth@gmail.com> | 2022-04-11 21:07:50 +0000 |
commit | 8442d63c5b662958c9ceb1bbf08bb492bf4a6d9b (patch) | |
tree | 39ecb799d69011c44574044b15d0f7f7946b886d /pkgs/development/python-modules/dm-haiku | |
parent | 158b9d526d7adebfaedab83604ab961c824edb33 (diff) |
python3Packages.dm-haiku: fix build
Diffstat (limited to 'pkgs/development/python-modules/dm-haiku')
-rw-r--r-- | pkgs/development/python-modules/dm-haiku/default.nix | 8 |
1 files changed, 7 insertions, 1 deletions
diff --git a/pkgs/development/python-modules/dm-haiku/default.nix b/pkgs/development/python-modules/dm-haiku/default.nix index 03677faa689fa..8a6eabfde0f56 100644 --- a/pkgs/development/python-modules/dm-haiku/default.nix +++ b/pkgs/development/python-modules/dm-haiku/default.nix @@ -21,7 +21,7 @@ buildPythonPackage rec { owner = "deepmind"; repo = pname; rev = "v${version}"; - sha256 = "sha256-qvKMeGPiWXvvyV+GZdTWdsC6Wp08AmP8nDtWk7sZtqM="; + hash = "sha256-qvKMeGPiWXvvyV+GZdTWdsC6Wp08AmP8nDtWk7sZtqM="; }; propagatedBuildInputs = [ @@ -32,6 +32,7 @@ buildPythonPackage rec { checkInputs = [ chex cloudpickle + dill dm-tree jaxlib pytest-xdist @@ -55,6 +56,11 @@ buildPythonPackage rec { "haiku/_src/integration/jax2tf_test.py" ]; + disabledTests = [ + # See https://github.com/deepmind/dm-haiku/issues/366. + "test_jit_Recurrent" + ]; + meta = with lib; { description = "Haiku is a simple neural network library for JAX developed by some of the authors of Sonnet."; homepage = "https://github.com/deepmind/dm-haiku"; |