diff options
Diffstat (limited to 'pkgs/development/python-modules/pyro-ppl/default.nix')
-rw-r--r-- | pkgs/development/python-modules/pyro-ppl/default.nix | 4 |
1 files changed, 2 insertions, 2 deletions
diff --git a/pkgs/development/python-modules/pyro-ppl/default.nix b/pkgs/development/python-modules/pyro-ppl/default.nix index 7bc6ac00925..b14119c3280 100644 --- a/pkgs/development/python-modules/pyro-ppl/default.nix +++ b/pkgs/development/python-modules/pyro-ppl/default.nix @@ -10,7 +10,7 @@ , pillow , pyro-api , pythonOlder -, pytorch +, torch , scikit-learn , seaborn , torchvision @@ -32,7 +32,7 @@ buildPythonPackage rec { propagatedBuildInputs = [ pyro-api - pytorch + torch networkx opt-einsum tqdm |