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
|
{
lib,
stdenv,
buildPythonPackage,
fetchFromGitHub,
fetchpatch,
pytestCheckHook,
setuptools,
apricot-select,
networkx,
numpy,
scikit-learn,
scipy,
torch,
}:
buildPythonPackage rec {
pname = "pomegranate";
version = "1.1.2";
pyproject = true;
src = fetchFromGitHub {
repo = "pomegranate";
owner = "jmschrei";
tag = "v${version}";
hash = "sha256-p2Gn0FXnsAHvRUeAqx4M1KH0+XvDl3fmUZZ7MiMvPSs=";
};
build-system = [ setuptools ];
dependencies = [
apricot-select
networkx
numpy
scikit-learn
scipy
torch
];
pythonImportsCheck = [ "pomegranate" ];
nativeCheckInputs = [
pytestCheckHook
];
patches = [
# Fix tests for pytorch 2.6
(fetchpatch {
name = "python-2.6.patch";
url = "https://github.com/jmschrei/pomegranate/pull/1142/commits/9ff5d5e2c959b44e569937e777b26184d1752a7b.patch";
hash = "sha256-BXsVhkuL27QqK/n6Fa9oJCzrzNcL3EF6FblBeKXXSts=";
})
];
pytestFlagsArray = lib.optionals (stdenv.hostPlatform.isDarwin && stdenv.hostPlatform.isx86_64) [
# AssertionError: Arrays are not almost equal to 6 decimals
"--deselect=tests/distributions/test_normal_full.py::test_fit"
"--deselect=tests/distributions/test_normal_full.py::test_from_summaries"
"--deselect=tests/distributions/test_normal_full.py::test_serialization"
];
disabledTests = [
# AssertionError: Arrays are not almost equal to 6 decimals
"test_sample"
];
meta = {
description = "Probabilistic and graphical models for Python, implemented in cython for speed";
homepage = "https://github.com/jmschrei/pomegranate";
changelog = "https://github.com/jmschrei/pomegranate/releases/tag/v${version}";
license = lib.licenses.mit;
maintainers = with lib.maintainers; [ rybern ];
};
}
|
