pythonPackages.tensorflow-estimator_1_15_1: init at 1.15.1

TF 1.15 still needs an older version of the tensorflow-estimator
package.

(cherry picked from commit c539f937c5)
This commit is contained in:
Andreas Rammhold 2020-03-02 17:22:35 +01:00 committed by John Ericson
parent ca6ccd4f4c
commit 07bd8938f3
3 changed files with 30 additions and 2 deletions

View file

@ -0,0 +1,27 @@
{ stdenv, fetchPypi, buildPythonPackage
, numpy
, absl-py
, mock
}:
buildPythonPackage rec {
pname = "tensorflow-estimator";
version = "1.15.1";
format = "wheel";
src = fetchPypi {
pname = "tensorflow_estimator";
inherit version format;
sha256 = "1fc61wmc0w22frs79j2x4g6wnv5g21xc6rix1g4bsvy9qfvvylw8";
};
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 ];
};
}

View file

@ -7,7 +7,7 @@
, future, setuptools, wheel, keras-preprocessing, keras-applications, google-pasta
, functools32
, opt-einsum
, termcolor, grpcio, six, wrapt, protobuf, tensorflow-estimator
, termcolor, grpcio, six, wrapt, protobuf, tensorflow-estimator_1_15_1
# Common deps
, git, swig, which, binutils, glibcLocales, cython
# Common libraries
@ -373,7 +373,7 @@ in buildPythonPackage {
numpy
six
protobuf
tensorflow-estimator
tensorflow-estimator_1_15_1
termcolor
wrapt
grpcio

View file

@ -6549,6 +6549,7 @@ in {
zerobin = callPackage ../development/python-modules/zerobin { };
tensorflow-estimator = callPackage ../development/python-modules/tensorflow-estimator { };
tensorflow-estimator_1_15_1 = callPackage ../development/python-modules/tensorflow-estimator/1_15_1.nix { };
tensorflow-probability = callPackage ../development/python-modules/tensorflow-probability { };