add option to build without Tensorflow

This commit is contained in:
Jethro Kuan 2018-11-08 22:06:13 +08:00 committed by Frederik Rietdijk
parent f11cd8165d
commit 78e557806e

View file

@ -5,6 +5,7 @@
, enum34
, absl-py
, tensorflow
, withTensorflow ? true
}:
buildPythonPackage rec {
@ -17,7 +18,7 @@ buildPythonPackage rec {
};
buildInputs = [ six enum34 tensorflow ];
propogatedBuildInputs = [ six enum34 ] + lib.optional withTensorflow tensorflow;
checkInputs = [ absl-py tensorflow ];
meta = with lib; {