tjh.dev
fork
Configure Feed
Select the types of activity you want to include in your feed.
nixpkgs mirror (for testing)
github.com/NixOS/nixpkgs
nix
fork
Configure Feed
Select the types of activity you want to include in your feed.
python310Packages.einops: disable failing tests
- add pythonImportsCheck
- adjust inputs
+42
-31
+42
-31
pkgs/development/python-modules/einops/default.nix
+42
-31
pkgs/development/python-modules/einops/default.nix
···
1
1
{ lib
2
-
, fetchFromGitHub
3
2
, buildPythonPackage
4
-
, numpy
5
-
, nose
6
-
, nbformat
7
-
, nbconvert
8
-
, jupyter
9
3
, chainer
10
-
, parameterized
11
-
, pytorch
12
-
, mxnet
13
-
, tensorflow
4
+
, fetchFromGitHub
5
+
, jupyter
14
6
, keras
7
+
#, mxnet
8
+
, nbconvert
9
+
, nbformat
10
+
, nose
11
+
, numpy
12
+
, parameterized
13
+
, pytestCheckHook
14
+
, pythonOlder
15
15
}:
16
16
17
17
buildPythonPackage rec {
18
18
pname = "einops";
19
19
version = "0.4.1";
20
+
format = "setuptools";
21
+
22
+
disabled = pythonOlder "3.7";
20
23
21
24
src = fetchFromGitHub {
22
25
owner = "arogozhnikov";
···
29
26
};
30
27
31
28
checkInputs = [
29
+
chainer
30
+
jupyter
31
+
keras
32
+
# mxnet (has issues with some CPUs, segfault)
33
+
nbconvert
34
+
nbformat
32
35
nose
33
36
numpy
34
-
# For notebook tests
35
-
nbformat
36
-
nbconvert
37
-
jupyter
38
-
# For backend tests
39
-
chainer
40
37
parameterized
41
-
pytorch
42
-
mxnet
43
-
tensorflow
44
-
keras
38
+
pytestCheckHook
45
39
];
46
40
47
41
# No CUDA in sandbox
48
42
EINOPS_SKIP_CUPY = 1;
49
43
50
-
checkPhase = ''
51
-
export HOME=$TMPDIR
52
-
53
-
# Prevent hangs on PyTorch-related tests, see
54
-
# https://discuss.pytorch.org/t/pytorch-cpu-hangs-on-nn-linear/17748/4
55
-
export OMP_NUM_THREADS=1
56
-
57
-
nosetests -v -w tests
44
+
preCheck = ''
45
+
export HOME=$(mktemp -d);
58
46
'';
59
47
60
-
meta = {
48
+
pythonImportsCheck = [
49
+
"einops"
50
+
];
51
+
52
+
disabledTests = [
53
+
# Tests are failing as mxnet is not pulled-in
54
+
# https://github.com/NixOS/nixpkgs/issues/174872
55
+
"test_all_notebooks"
56
+
"test_dl_notebook_with_all_backends"
57
+
"test_backends_installed"
58
+
];
59
+
60
+
disabledTestPaths = [
61
+
"tests/test_layers.py"
62
+
];
63
+
64
+
meta = with lib; {
61
65
description = "Flexible and powerful tensor operations for readable and reliable code";
62
66
homepage = "https://github.com/arogozhnikov/einops";
63
-
license = lib.licenses.mit;
64
-
maintainers = with lib.maintainers; [ yl3dy ];
67
+
license = licenses.mit;
68
+
maintainers = with maintainers; [ yl3dy ];
65
69
};
66
70
}
71
+