diff options
Diffstat (limited to 'pkgs/development/python-modules/tensorflow-estimator/2/default.nix')
-rw-r--r-- | pkgs/development/python-modules/tensorflow-estimator/2/default.nix | 27 |
1 files changed, 27 insertions, 0 deletions
diff --git a/pkgs/development/python-modules/tensorflow-estimator/2/default.nix b/pkgs/development/python-modules/tensorflow-estimator/2/default.nix new file mode 100644 index 00000000000..47ce0f83b02 --- /dev/null +++ b/pkgs/development/python-modules/tensorflow-estimator/2/default.nix @@ -0,0 +1,27 @@ +{ stdenv, fetchPypi, buildPythonPackage +, numpy +, absl-py +, mock +}: + +buildPythonPackage rec { + pname = "tensorflow-estimator"; + version = "2.1.0"; + format = "wheel"; + + src = fetchPypi { + pname = "tensorflow_estimator"; + inherit version format; + sha256 = "0wk9viil54ms1s2ir7zxygqa425i69hx8zngwhdqvw9nlr4gdig5"; + }; + + propagatedBuildInputs = [ mock numpy absl-py ]; + + meta = with stdenv.lib; { + description = "TensorFlow Estimator is a high-level API that encapsulates model training, evaluation, prediction, and exporting."; + homepage = http://tensorflow.org; + license = licenses.asl20; + maintainers = with maintainers; [ jyp ]; + }; +} + |