tjh.dev
nixpkgs mirror (for testing)
github.com/NixOS/nixpkgs
nix
python312Packages.distrax: adapt to jax 0.6.0 new API changes
+14
+14
pkgs/development/python-modules/distrax/default.nix
+14
pkgs/development/python-modules/distrax/default.nix
···
2
2
lib,
3
3
buildPythonPackage,
4
4
fetchFromGitHub,
5
+
fetchpatch,
5
6
chex,
6
7
jaxlib,
7
8
numpy,
···
23
22
tag = "v${version}";
24
23
hash = "sha256-A1aCL/I89Blg9sNmIWQru4QJteUTN6+bhgrEJPmCrM0=";
25
24
};
25
+
26
+
patches = [
27
+
# TODO: remove at the next release (already on master)
28
+
(fetchpatch {
29
+
name = "fix-jax-0.6.0-compat";
30
+
url = "https://github.com/google-deepmind/distrax/commit/c02708ac46518fac00ab2945311e0f2ee32c672c.patch";
31
+
hash = "sha256-hFNXKoA1b5I6dzhwTRXp/SnkHv89GI6tYwlnBBHwG78=";
32
+
})
33
+
];
26
34
27
35
dependencies = [
28
36
chex
···
81
71
];
82
72
83
73
disabledTestPaths = [
74
+
# Since jax 0.6.0:
75
+
# TypeError: <lambda>() got an unexpected keyword argument 'accuracy'
76
+
"distrax/_src/bijectors/lambda_bijector_test.py"
77
+
84
78
# TypeErrors
85
79
"distrax/_src/bijectors/tfp_compatible_bijector_test.py"
86
80
"distrax/_src/distributions/distribution_from_tfp_test.py"