tjh.dev
1{ stdenv, fetchPypi, buildPythonPackage
2, numpy
3, absl-py
4, mock
5}:
6
7buildPythonPackage rec {
8 pname = "tensorflow-estimator";
9 # This is effectively 1.15.0. Upstream tagged 1.15.0 by mistake before
10 # actually updating the version in setup.py, which is why this tag is called
11 # 1.15.1.
12 version = "1.15.1";
13 format = "wheel";
14
15 src = fetchPypi {
16 pname = "tensorflow_estimator";
17 inherit version format;
18 sha256 = "1fc61wmc0w22frs79j2x4g6wnv5g21xc6rix1g4bsvy9qfvvylw8";
19 };
20
21 propagatedBuildInputs = [ mock numpy absl-py ];
22
23 meta = with stdenv.lib; {
24 description = "TensorFlow Estimator is a high-level API that encapsulates model training, evaluation, prediction, and exporting.";
25 homepage = http://tensorflow.org;
26 license = licenses.asl20;
27 maintainers = with maintainers; [ jyp ];
28 };
29}
30