about summary refs log tree commit diff
path: root/pkgs/development/python-modules/dm-haiku
diff options
context:
space:
mode:
authorAlexander Tsvyashchenko <ndl@endl.ch>2022-04-13 23:03:13 +0200
committerAlexander Tsvyashchenko <ndl@endl.ch>2022-04-13 23:03:13 +0200
commitd580e84c541992aacc7974666f29f68ef29cff47 (patch)
tree21c45e0c2b3db7df9ae4683a17aac31a7d09bb5e /pkgs/development/python-modules/dm-haiku
parenta2a665fad5e9ac86fa63e703e09b1477164663cc (diff)
Fixed `dm-haiku` and `flax` builds.
Diffstat (limited to 'pkgs/development/python-modules/dm-haiku')
-rw-r--r--pkgs/development/python-modules/dm-haiku/default.nix2
1 files changed, 2 insertions, 0 deletions
diff --git a/pkgs/development/python-modules/dm-haiku/default.nix b/pkgs/development/python-modules/dm-haiku/default.nix
index 8a6eabfde0f56..804233804a1fe 100644
--- a/pkgs/development/python-modules/dm-haiku/default.nix
+++ b/pkgs/development/python-modules/dm-haiku/default.nix
@@ -54,6 +54,8 @@ buildPythonPackage rec {
     # likely that's the reason the upstream uses TF-nightly for tests?
     # `nixpkgs` doesn't have the corresponding TF version packaged.
     "haiku/_src/integration/jax2tf_test.py"
+    # `TypeError: lax.conv_general_dilated requires arguments to have the same dtypes, got float32, float16`.
+    "haiku/_src/integration/numpy_inputs_test.py"
   ];
 
   disabledTests = [