about summary refs log tree commit diff
path: root/pkgs/applications/science
diff options
context:
space:
mode:
authoravitex <theavitex@gmail.com>2019-06-17 22:16:13 +1000
committerFrederik Rietdijk <freddyrietdijk@fridh.nl>2019-07-20 13:17:17 +0200
commit44324bdd7c88f043b5c1af084dd34f955017a304 (patch)
treeeea7a94790f2aa710c30ab7e7de17585d71e12d2 /pkgs/applications/science
parent955ed925e56c3b18fd378da5a1e2b8741f32335e (diff)
downloadnixlib-44324bdd7c88f043b5c1af084dd34f955017a304.tar
nixlib-44324bdd7c88f043b5c1af084dd34f955017a304.tar.gz
nixlib-44324bdd7c88f043b5c1af084dd34f955017a304.tar.bz2
nixlib-44324bdd7c88f043b5c1af084dd34f955017a304.tar.lz
nixlib-44324bdd7c88f043b5c1af084dd34f955017a304.tar.xz
nixlib-44324bdd7c88f043b5c1af084dd34f955017a304.tar.zst
nixlib-44324bdd7c88f043b5c1af084dd34f955017a304.zip
caffe: add python package requirements
Diffstat (limited to 'pkgs/applications/science')
-rw-r--r--pkgs/applications/science/math/caffe/default.nix14
1 files changed, 11 insertions, 3 deletions
diff --git a/pkgs/applications/science/math/caffe/default.nix b/pkgs/applications/science/math/caffe/default.nix
index 2a736d3d398e..153ebaa441f5 100644
--- a/pkgs/applications/science/math/caffe/default.nix
+++ b/pkgs/applications/science/math/caffe/default.nix
@@ -73,9 +73,17 @@ stdenv.mkDerivation rec {
                 ++ lib.optionals stdenv.isDarwin [ Accelerate CoreGraphics CoreVideo ]
                 ;
 
-  propagatedBuildInputs = lib.optional pythonSupport python.pkgs.protobuf;
-
-  outputs = [ "bin" "out"];
+  propagatedBuildInputs = lib.optionals pythonSupport (
+    # requirements.txt
+    let pp = python.pkgs; in ([
+      pp.numpy pp.scipy pp.scikitimage pp.h5py
+      pp.matplotlib pp.ipython pp.networkx pp.nose
+      pp.pandas pp.dateutil pp.protobuf pp.gflags
+      pp.pyyaml pp.pillow pp.six
+    ] ++ lib.optional leveldbSupport pp.leveldb)
+  );
+
+  outputs = [ "bin" "out" ];
   propagatedBuildOutputs = []; # otherwise propagates out -> bin cycle
 
   patches = [