nordic-dev.net/nixpkgs
nordic-dev.net/nixpkgs
2
0
Fork 0
mirror of https://github.com/NixOS/nixpkgs.git synced 2025-02-16 09:03:42 +00:00
Code Issues Packages Projects Releases Wiki Activity

Merge pull request #168545 from ndl/master

Browse Source 
Fixed `dm-haiku` and `flax` builds
This commit is contained in:
Samuel Ainsworth 2022-04-16 12:26:18 -07:00 committed by GitHub
commit 4c0422397f
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
2 changed files with 5 additions and 3 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

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

@ -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 = [

6
pkgs/development/python-modules/flax/default.nix
View File

@ -14,13 +14,13 @@
buildPythonPackage rec {
pname = "flax";
version = "0.4.0";
version = "0.4.1";
src = fetchFromGitHub {
owner = "google";
repo = pname;
rev = "v${version}";
sha256 = "0rvdaxyf68qmm5d77gbizpcibyz2ic2pb2x7rgf7p8qwijyc39ws";
sha256 = "0j5ngdndm9nm49gcda7m36qzwk5lcbi4jnij9fi96vld54ip6f6v";
};
buildInputs = [ jaxlib ];
@ -42,7 +42,7 @@ buildPythonPackage rec {
pytestCheckHook
tensorflow
];
pytestFlagsArray = [ "-n $NIX_BUILD_CORES" ];
pytestFlagsArray = [ "-n $NIX_BUILD_CORES -W ignore::FutureWarning" ];
disabledTestPaths = [
# Docs test, needs extra deps + we're not interested in it.