Merge pull request #52936 from teh/pyro

Pyro
2018-12-31 03:47:51 +01:00
parent f43c75e37c da01f2a745
commit a02b6bc318
3 changed files with 73 additions and 0 deletions
--- a/pkgs/top-level/python-packages.nix
+++ b/pkgs/top-level/python-packages.nix
@@ -2423,6 +2423,10 @@ in {
    cudaSupport = pkgs.config.cudaSupport or false;
  };

+  pyro-ppl = callPackage ../development/python-modules/pyro-ppl {};
+
+  opt-einsum = callPackage ../development/python-modules/opt-einsum {};
+
  pytorchWithCuda = self.pytorch.override {
    cudaSupport = true;
  };