diff options
Diffstat (limited to 'nixpkgs/pkgs/development/python-modules/einops/default.nix')
-rw-r--r-- | nixpkgs/pkgs/development/python-modules/einops/default.nix | 2 |
1 files changed, 2 insertions, 0 deletions
diff --git a/nixpkgs/pkgs/development/python-modules/einops/default.nix b/nixpkgs/pkgs/development/python-modules/einops/default.nix index 34510762abf7..a2de3a70971d 100644 --- a/nixpkgs/pkgs/development/python-modules/einops/default.nix +++ b/nixpkgs/pkgs/development/python-modules/einops/default.nix @@ -10,6 +10,7 @@ , pytorch , mxnet , tensorflow +, keras }: buildPythonPackage rec { @@ -35,6 +36,7 @@ buildPythonPackage rec { pytorch mxnet tensorflow + keras ]; # No CUDA in sandbox |