From 1d00bb137598b3bc99ff473b139f1f7878c8c43f Mon Sep 17 00:00:00 2001 From: Timo Kaufmann Date: Fri, 12 Apr 2019 17:48:57 +0200 Subject: [PATCH] python.pkgs.tensorflow-probability: init at 0.6.0 Needed as a dependency for dm-sonnet. --- .../tensorflow-probability/default.nix | 41 +++++++++++++++++++ pkgs/top-level/python-packages.nix | 2 + 2 files changed, 43 insertions(+) create mode 100644 pkgs/development/python-modules/tensorflow-probability/default.nix diff --git a/pkgs/development/python-modules/tensorflow-probability/default.nix b/pkgs/development/python-modules/tensorflow-probability/default.nix new file mode 100644 index 00000000000..aa32904b8d3 --- /dev/null +++ b/pkgs/development/python-modules/tensorflow-probability/default.nix @@ -0,0 +1,41 @@ +{ lib +, fetchFromGitHub +, buildPythonPackage +, tensorflow +, pytest +}: + +buildPythonPackage rec { + pname = "tensorflow-probability"; + version = "0.6.0"; + + src = fetchFromGitHub { + owner = "tensorflow"; + repo = "probability"; + rev = "v${version}"; + sha256 = "1y210n4asv8j39pk68bdfrz01gddflvzhxbcvj5jw6rjgaagnhvx"; + }; + + propagatedBuildInputs = [ + tensorflow + ]; + + checkInputs = [ + pytest + ]; + + # Tests have an invalid import (`tensorflow_probability.opensource`), should + # be resolved in the next version with + # https://github.com/tensorflow/probability/commit/77d5957f2f0bdddcb46582799cd9c5c5167a1a40 + doCheck = false; + checkPhase = '' + py.test + ''; + + meta = with lib; { + description = "Library for probabilistic reasoning and statistical analysis"; + homepage = https://www.tensorflow.org/probability/; + license = licenses.asl20; + maintainers = with maintainers; [ timokau ]; + }; +} diff --git a/pkgs/top-level/python-packages.nix b/pkgs/top-level/python-packages.nix index 0bf2158243c..1de7388833e 100644 --- a/pkgs/top-level/python-packages.nix +++ b/pkgs/top-level/python-packages.nix @@ -5143,6 +5143,8 @@ in { tensorflow-estimator = callPackage ../development/python-modules/tensorflow-estimator { }; + tensorflow-probability = callPackage ../development/python-modules/tensorflow-probability { }; + tensorflow-tensorboard = callPackage ../development/python-modules/tensorflow-tensorboard { }; tensorflow =