Skip to content

Commit

Permalink
pythonPackages.tensorflow-estimator_1_15_1: init at 1.15.1
Browse files Browse the repository at this point in the history
TF 1.15 still needs an older version of the tensorflow-estimator
package.
  • Loading branch information
andir committed Mar 3, 2020
1 parent 84b3678 commit c539f93
Show file tree
Hide file tree
Showing 3 changed files with 30 additions and 2 deletions.
27 changes: 27 additions & 0 deletions pkgs/development/python-modules/tensorflow-estimator/1_15_1.nix
@@ -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 ];
};
}

4 changes: 2 additions & 2 deletions pkgs/development/python-modules/tensorflow/default.nix
Expand Up @@ -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
Expand Down Expand Up @@ -374,7 +374,7 @@ in buildPythonPackage {
numpy
six
protobuf
tensorflow-estimator
tensorflow-estimator_1_15_1
termcolor
wrapt
grpcio
Expand Down
1 change: 1 addition & 0 deletions pkgs/top-level/python-packages.nix
Expand Up @@ -6657,6 +6657,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 { };

Expand Down

0 comments on commit c539f93

Please sign in to comment.