about summary refs log tree commit diff
path: root/pkgs/development/python-modules/dm-sonnet
diff options
context:
space:
mode:
authorTimo Kaufmann <timokau@zoho.com>2019-06-21 18:23:50 +0200
committerTimo Kaufmann <timokau@zoho.com>2019-06-26 23:48:33 +0200
commitc393b601797ac5d5038ab45d9ce9bab47947ce10 (patch)
treeeeae5358152cbc9b0fc3262567c294933b42f9cd /pkgs/development/python-modules/dm-sonnet
parentc2948f640bd5c5d646ec37c72e91e90486db49c7 (diff)
python.pkgs.dm-sonnet: add estimator dependency
Previously was implicitly included in the binary tensorflow release.
Diffstat (limited to 'pkgs/development/python-modules/dm-sonnet')
-rw-r--r--pkgs/development/python-modules/dm-sonnet/default.nix2
1 files changed, 2 insertions, 0 deletions
diff --git a/pkgs/development/python-modules/dm-sonnet/default.nix b/pkgs/development/python-modules/dm-sonnet/default.nix
index 1c972cb2aa020..41dc30e471906 100644
--- a/pkgs/development/python-modules/dm-sonnet/default.nix
+++ b/pkgs/development/python-modules/dm-sonnet/default.nix
@@ -11,6 +11,7 @@
 , wrapt
 , tensorflow
 , tensorflow-probability
+, tensorflow-estimator
 }:
 
 let
@@ -73,6 +74,7 @@ in buildPythonPackage rec {
     wrapt
     tensorflow
     tensorflow-probability
+    tensorflow-estimator
   ];
 
   # not sure how to properly run the real test suite -- through bazel?