From 1997ca354eb7ef5775462d331ee595939053a5d3 Mon Sep 17 00:00:00 2001 From: Jean-Philippe Bernardy Date: Thu, 28 Feb 2019 11:57:08 +0100 Subject: [PATCH] pythonPackages.tensorflow-estimator: init 1.13.0 --- .../tensorflow-estimator/default.nix | 27 +++++++++++++++++++ pkgs/top-level/python-packages.nix | 2 ++ 2 files changed, 29 insertions(+) create mode 100644 pkgs/development/python-modules/tensorflow-estimator/default.nix diff --git a/pkgs/development/python-modules/tensorflow-estimator/default.nix b/pkgs/development/python-modules/tensorflow-estimator/default.nix new file mode 100644 index 00000000000..5b9e032c49a --- /dev/null +++ b/pkgs/development/python-modules/tensorflow-estimator/default.nix @@ -0,0 +1,27 @@ +{ stdenv, lib, fetchPypi, buildPythonPackage, isPy3k +, numpy +, absl-py +, mock +}: + +buildPythonPackage rec { + pname = "tensorflow-estimator"; + version = "1.13.0"; + format = "wheel"; + + src = fetchPypi { + pname = "tensorflow_estimator"; + inherit version format; + sha256 = "068l4w0w7dj9gqkf8avjclq9zsp7ifwzw4rpf4qjylz3hczamzbw"; + }; + + 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 ]; + }; +} + diff --git a/pkgs/top-level/python-packages.nix b/pkgs/top-level/python-packages.nix index 0fbe97886ae..b45f230628f 100644 --- a/pkgs/top-level/python-packages.nix +++ b/pkgs/top-level/python-packages.nix @@ -5054,6 +5054,8 @@ in { zerobin = callPackage ../development/python-modules/zerobin { }; + tensorflow-estimator = callPackage ../development/python-modules/tensorflow-estimator { }; + tensorflow-tensorboard = callPackage ../development/python-modules/tensorflow-tensorboard { }; tensorflow = disabledIf isPy37 (