python.pkgs.dm-sonnet: add estimator dependency

Previously was implicitly included in the binary tensorflow release.
2019-06-21 18:23:50 +02:00 · 2019-06-21 18:23:50 +02:00 · c393b60179
commit c393b60179
parent c2948f640b
1 changed files with 2 additions and 0 deletions
--- 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?