1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
|
{ lib
, autograd
, buildPythonPackage
, cupy
, cvxopt
, cython
, fetchPypi
, matplotlib
, numpy
, tensorflow
, pymanopt
, pytestCheckHook
, pythonOlder
, scikit-learn
, scipy
, enableDimensionalityReduction ? false
, enableGPU ? false
}:
buildPythonPackage rec {
pname = "pot";
version = "0.8.2";
format = "setuptools";
disabled = pythonOlder "3.6";
src = fetchPypi {
pname = "POT";
inherit version;
sha256 = "sha256-PKmuPI83DPy7RkOgHHPdPJJz5NT/fpr123AVTzTLwgQ=";
};
nativeBuildInputs = [
numpy
cython
];
propagatedBuildInputs = [
numpy
scipy
] ++ lib.optionals enableGPU [
cupy
] ++ lib.optionals enableDimensionalityReduction [
autograd
pymanopt
];
checkInputs = [
cvxopt
matplotlib
numpy
tensorflow
scikit-learn
pytestCheckHook
];
postPatch = ''
substituteInPlace setup.cfg \
--replace " --cov-report= --cov=ot" "" \
--replace " --durations=20" "" \
--replace " --junit-xml=junit-results.xml" ""
substituteInPlace setup.py \
--replace '"oldest-supported-numpy", ' ""
# we don't need setup.py to find the macos sdk for us
sed -i '/sdk_path/d' setup.py
'';
# To prevent importing of an incomplete package from the build directory
# instead of nix store (`ot` is the top-level package name).
preCheck = ''
rm -r ot
'';
disabledTests = [
# GPU tests are always skipped because of sandboxing
"warnings"
# Fixture is not available
"test_conditional_gradient"
"test_convert_between_backends"
"test_emd_backends"
"test_emd_emd2_types_devices"
"test_emd1d_type_devices"
"test_emd2_backends"
"test_factored_ot_backends"
"test_free_support_barycenter_backends"
"test_func_backends"
"test_generalized_conditional_gradient"
"test_line_search_armijo"
"test_loss_dual"
"test_max_sliced_backend"
"test_plan_dual"
"test_random_backends"
"test_sliced_backend"
"test_to_numpy"
"test_wasserstein_1d_type_devices"
"test_wasserstein"
"test_weak_ot_bakends"
# TypeError: Only integers, slices...
"test_emd1d_device_tf"
];
disabledTestPaths = [
# AttributeError: module pytest has no attribute skip_backend
"test/test_bregman.py"
"test/test_da.py"
"test/test_utils.py"
"test/test_gromov.py"
"test/test_helpers.py"
"test/test_unbalanced.py"
] ++ lib.optionals (!enableDimensionalityReduction) [
"test/test_dr.py"
];
pythonImportsCheck = [
"ot"
"ot.lp"
];
meta = with lib; {
description = "Python Optimal Transport Library";
homepage = "https://pythonot.github.io/";
license = licenses.mit;
maintainers = with maintainers; [ yl3dy ];
};
}
|