{ stdenv, lib , buildPythonPackage, fetchPypi, isPy3k , filelock, protobuf, numpy, pytest, mock , cupy, cudaSupport ? false }: buildPythonPackage rec { pname = "chainer"; version = "4.3.1"; src = fetchPypi { inherit pname version; sha256 = "d56b769720ef9de93d5b74faea96729b076beb5362f97565654de3c723f2c822"; }; checkInputs = [ pytest mock ]; propagatedBuildInputs = [ filelock protobuf numpy ] ++ lib.optionals cudaSupport [ cupy ]; # In python3, test was failed... doCheck = !isPy3k; meta = with stdenv.lib; { description = "A flexible framework of neural networks for deep learning"; homepage = https://chainer.org/; license = licenses.mit; maintainers = with maintainers; [ hyphon81 ]; }; }