tjh.dev
nixpkgs mirror (for testing)
github.com/NixOS/nixpkgs
nix
python3Packages.flax: 0.6.0 -> 0.6.1
+18
-12
+18
-12
pkgs/development/python-modules/flax/default.nix
+18
-12
pkgs/development/python-modules/flax/default.nix
···
17
17
18
18
buildPythonPackage rec {
19
19
pname = "flax";
20
-
version = "0.6.0";
20
+
version = "0.6.1";
21
21
22
22
src = fetchFromGitHub {
23
23
owner = "google";
24
24
repo = pname;
25
25
rev = "refs/tags/v${version}";
26
-
sha256 = "sha256-egTYYFZxhE/Kk7jXRi1HmjCjyFia2LoRigH042isDu0=";
26
+
sha256 = "sha256-fZiODo+izOwGjCCTvi11GvUG/VQL1DV9bNXKjvIIw4A=";
27
27
};
28
-
29
-
patches = [
30
-
# Bump rich dependency, should be fixed in releases after 0.6.0
31
-
# https://github.com/google/flax/pull/2407
32
-
(fetchpatch {
33
-
url = "https://github.com/google/flax/commit/72189153f9779022b97858ae747c23fbaf571e3d.patch";
34
-
sha256 = "sha256-hKOn/M7qpBM6R1RIJpnXpRoZgIHqkwQZApN4L0fBzIE=";
35
-
name = "bump_rich_dependency.patch";
36
-
})
37
-
];
38
28
39
29
buildInputs = [ jaxlib ];
40
30
···
64
74
# `tensorflow_datasets`, `vocabulary`) so the benefits of trying to run them
65
75
# would be limited anyway.
66
76
"examples/*"
77
+
];
78
+
79
+
disabledTests = [
80
+
# See https://github.com/google/flax/issues/2554.
81
+
"test_async_save_checkpoints"
82
+
"test_jax_array0"
83
+
"test_jax_array1"
84
+
"test_keep0"
85
+
"test_keep1"
86
+
"test_optimized_lstm_cell_matches_regular"
87
+
"test_overwrite_checkpoints"
88
+
"test_save_restore_checkpoints_target_empty"
89
+
"test_save_restore_checkpoints_target_none"
90
+
"test_save_restore_checkpoints_target_singular"
91
+
"test_save_restore_checkpoints_w_float_steps"
92
+
"test_save_restore_checkpoints"
67
93
];
68
94
69
95
meta = with lib; {