pkgs/development/python-modules/mxnet/default.nix

   1 { lib
   2 , buildPythonPackage
   3 , pkgs
   4 , requests
   5 , numpy
   6 , graphviz
   7 , python
   8 , isPy3k
   9 }:
  10
  11 buildPythonPackage {
  12   inherit (pkgs.mxnet) pname version src meta;
  13
  14   buildInputs = [ pkgs.mxnet ];
  15   propagatedBuildInputs = [ requests numpy graphviz ];
  16
  17   LD_LIBRARY_PATH = lib.makeLibraryPath [ pkgs.mxnet ];
  18
  19   doCheck = !isPy3k;
  20
  21   postPatch = ''
  22     substituteInPlace python/setup.py \
  23       --replace "graphviz<0.9.0," "graphviz"
  24   '';
  25
  26   preConfigure = ''
  27     cd python
  28   '';
  29
  30   postInstall = ''
  31     rm -rf $out/mxnet
  32     ln -s ${pkgs.mxnet}/lib/libmxnet.so $out/${python.sitePackages}/mxnet
  33   '';
  34
  35 }