blob: e943d67c7baa2edb9afaca24cad6f5ef583678f2 (
plain) (
blame)
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
|
{ lib
, buildPythonPackage
, fetchpatch
, fetchPypi
, setuptools
, numpy
, pytestCheckHook
, pythonOlder
}:
buildPythonPackage rec {
version = "3.3.0";
pname = "opt-einsum";
pyproject = true;
disabled = pythonOlder "3.6";
src = fetchPypi {
pname = "opt_einsum";
inherit version;
hash = "sha256-WfZHX3e7w33PfNdIUZwOxgci6R5jyhFOaIIcDFSkZUk=";
};
patches = [
# https://github.com/dgasmith/opt_einsum/pull/208
(fetchpatch {
name = "python312-compatibility.patch";
url = "https://github.com/dgasmith/opt_einsum/commit/0beacf96923bbb2dd1939a9c59398a38ce7a11b1.patch";
hash = "sha256-dmmEzhy17huclo1wOubpBUDc2L7vqEU5b/6a5loM47A=";
})
];
nativeBuildInputs = [
setuptools
];
propagatedBuildInputs = [
numpy
];
nativeCheckInputs = [
pytestCheckHook
];
pythonImportsCheck = [
"opt_einsum"
];
meta = with lib; {
description = "Optimizing NumPy's einsum function with order optimization and GPU support";
homepage = "https://github.com/dgasmith/opt_einsum";
license = licenses.mit;
maintainers = with maintainers; [ teh ];
};
}
|