alioth/nixpkgs
3
0
Fork 0
forked from mirrors/nixpkgs
Code Releases Activity

python3Packages.objax: fix tensorboard dependency (#156909)

Browse source 
Also moved `jaxlib` to `buildInputs` to avoid injecting it into dependent packages.
This commit is contained in:
Alexander Tsvyashchenko 2022-01-27 20:45:32 +01:00 committed by GitHub
parent e5808c2f62
commit fa7b83fa48
No known key found for this signature in database
GPG key ID: 4AEE18F83AFDEB23
1 changed files with 8 additions and 5 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

13
pkgs/development/python-modules/objax/default.nix
View file

@ -7,7 +7,7 @@
, parameterized
, pillow
, scipy
, tensorflow-tensorboard_2 ? null
, tensorflow-tensorboard
}:
buildPythonPackage rec {
@ -21,14 +21,19 @@ buildPythonPackage rec {
sha256 = "09gm61ghn5mi92q5mhx22mcv6aa6z78jsrnfar1hd3nwwyn9dq42";
};
# Avoid propagating the dependency on `jaxlib`, see
# https://github.com/NixOS/nixpkgs/issues/156767
buildInputs = [
jaxlib
];
propagatedBuildInputs = [
jax
jaxlib
numpy
parameterized
pillow
scipy
tensorflow-tensorboard_2
tensorflow-tensorboard
];
pythonImportsCheck = [
@ -40,7 +45,5 @@ buildPythonPackage rec {
homepage = "https://github.com/google/objax";
license = licenses.asl20;
maintainers = with maintainers; [ ndl ];
# Darwin doesn't have `tensorflow-tensorboard_2` which is required by wheel deps.
platforms = [ "aarch64-linux" "x86_64-linux" ];
};
}