python.packages.tensorflow: add missing dependencies

This commit is contained in:
Jean-Philippe Bernardy 2018-06-07 13:30:29 +02:00
parent 1477171073
commit 7d02ec9ac6

View file

@ -3,8 +3,11 @@
, fetchurl
, buildPythonPackage
, isPy3k, isPy35, isPy36, pythonOlder
, astor
, gast
, numpy
, six
, termcolor
, protobuf
, absl-py
, mock
@ -47,7 +50,7 @@ in buildPythonPackage rec {
dls = import ./tf1.7.1-hashes.nix;
in fetchurl dls.${key};
propagatedBuildInputs = [ numpy six protobuf absl-py ]
propagatedBuildInputs = [ numpy six protobuf absl-py astor gast termcolor ]
++ lib.optional (!isPy3k) mock
++ lib.optionals (pythonOlder "3.4") [ backports_weakref enum34 ]
++ lib.optional (pythonOlder "3.6") tensorflow-tensorboard;