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