add option to build without Tensorflow

This commit is contained in:
Jethro Kuan
2018-11-11 08:39:19 +01:00
committed by Frederik Rietdijk
parent f11cd8165d
commit 78e557806e
@@ -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; {