pyrox.dev
1{ lib, buildPythonPackage, fetchFromGitHub, isPy3k
2, filelock, protobuf, numpy, pytestCheckHook, mock, typing-extensions
3, cupy, cudaSupport ? false
4}:
5
6buildPythonPackage rec {
7 pname = "chainer";
8 version = "7.8.0";
9 disabled = !isPy3k; # python2.7 abandoned upstream
10
11 # no tests in Pypi tarball
12 src = fetchFromGitHub {
13 owner = "chainer";
14 repo = "chainer";
15 rev = "v${version}";
16 sha256 = "1zfj3pk54gzxd4nid0qjx4kw1wdngwscvn4hk4cijxvwqi4a5zxj";
17 };
18
19 checkInputs = [
20 pytestCheckHook
21 mock
22 ];
23
24 propagatedBuildInputs = [
25 filelock
26 protobuf
27 numpy
28 typing-extensions
29 ] ++ lib.optionals cudaSupport [ cupy ];
30
31 pytestFlagsArray = [ "tests/chainer_tests/utils_tests" ];
32
33 disabledTests = [
34 "gpu"
35 "cupy"
36 "ideep"
37 ];
38
39 meta = with lib; {
40 description = "A flexible framework of neural networks for deep learning";
41 homepage = "https://chainer.org/";
42 license = licenses.mit;
43 maintainers = with maintainers; [ hyphon81 ];
44 };
45}