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