pyrox.dev
1{
2 lib,
3 fetchPypi,
4 buildPythonPackage,
5 numpy,
6 absl-py,
7 mock,
8}:
9
10buildPythonPackage rec {
11 pname = "tensorflow-estimator";
12 version = "2.15.0";
13 format = "wheel";
14
15 src = fetchPypi {
16 pname = "tensorflow_estimator";
17 inherit version format;
18 hash = "sha256-rt8h7sf7LckRUPyRoc4SvETbtyJ4oItY55/4fJ4o8VM=";
19 };
20
21 propagatedBuildInputs = [
22 mock
23 numpy
24 absl-py
25 ];
26
27 meta = with lib; {
28 description = "TensorFlow Estimator is a high-level API that encapsulates model training, evaluation, prediction, and exporting";
29 homepage = "http://tensorflow.org";
30 license = licenses.asl20;
31 };
32}