···

       
3
       
 
       
  stdenv,

     

       
4
       
 
       
  buildPythonPackage,

     

       
5
       
 
       
  fetchFromGitHub,

     

       
6
       
-
       


     

       
7
       
-
       
  # build-system

     

       
8
       
 
       
  setuptools,

     

       
9
       
-
       


     

       
10
       
-
       
  # dependencies

     

       
11
       
 
       
  apricot-select,

     

       
12
       
 
       
  networkx,

     

       
13
       
 
       
  numpy,

     

       
14
       
 
       
  scikit-learn,

     

       
15
       
 
       
  scipy,

     

       
16
       
 
       
  torch,

     

       
17
       
-
       


     

       
18
       
-
       
  # tests

     

       
19
       
-
       
  pytestCheckHook,

     

       
20
       
 
       
}:

     

       
21
       
 
       


     

       
22
       
 
       
buildPythonPackage rec {

     
···

       
27
       
 
       
  src = fetchFromGitHub {

     

       
28
       
 
       
    repo = "pomegranate";

     

       
29
       
 
       
    owner = "jmschrei";

     

       
30
       
-
       
    # tag = "v${version}";

     

       
31
       
-
       
    # No tag for 1.1.2

     

       
32
       
-
       
    rev = "e9162731f4f109b7b17ecffde768734cacdb839b";

     

       
33
       
-
       
    hash = "sha256-vVoAoZ+mph11ZfINT+yxRyk9rXv6FBDgxBz56P2K95Y=";

     

       
34
       
 
       
  };

     

       
35
       
 
       


     

       
36
       
-
       
  # _pickle.UnpicklingError: Weights only load failed.

     

       
37
       
-
       
  # https://pytorch.org/docs/stable/generated/torch.load.html

     

       
38
       
-
       
  postPatch = ''

     

       
39
       
-
       
    substituteInPlace \

     

       
40
       
-
       
      tests/distributions/test_bernoulli.py \

     

       
41
       
-
       
      tests/distributions/test_categorical.py \

     

       
42
       
-
       
      tests/distributions/test_exponential.py \

     

       
43
       
-
       
      tests/distributions/test_gamma.py \

     

       
44
       
-
       
      tests/distributions/test_independent_component.py \

     

       
45
       
-
       
      tests/distributions/test_normal_diagonal.py \

     

       
46
       
-
       
      tests/distributions/test_normal_full.py \

     

       
47
       
-
       
      tests/distributions/test_poisson.py \

     

       
48
       
-
       
      tests/distributions/test_student_t.py \

     

       
49
       
-
       
      tests/distributions/test_uniform.py \

     

       
50
       
-
       
      tests/test_bayes_classifier.py \

     

       
51
       
-
       
      tests/test_gmm.py \

     

       
52
       
-
       
      tests/test_kmeans.py \

     

       
53
       
-
       
      --replace-fail \

     

       
54
       
-
       
        'torch.load(".pytest.torch")' \

     

       
55
       
-
       
        'torch.load(".pytest.torch", weights_only=False)'

     

       
56
       
-
       
  '';

     

       
57
       
-
       


     

       
58
       
 
       
  build-system = [ setuptools ];

     

       
59
       
 
       


     

       
60
       
 
       
  dependencies = [

     
···

       
72
       
 
       
    pytestCheckHook

     

       
73
       
 
       
  ];

     

       
74
       
 
       


     

       
75
       
-
       
  disabledTestPaths = lib.optionals (stdenv.hostPlatform.isDarwin && stdenv.hostPlatform.isx86_64) [

     

       
0
       
 
       

     

       
0
       
 
       

     

       
0
       
 
       

     

       
0
       
 
       

     

       
0
       
 
       

     

       
0
       
 
       

     

       
0
       
 
       

     

       
0
       
 
       

     

       
0
       
 
       

     

       
76
       
 
       
    # AssertionError: Arrays are not almost equal to 6 decimals

     

       
77
       
-
       
    "=tests/distributions/test_normal_full.py::test_fit"

     

       
78
       
-
       
    "=tests/distributions/test_normal_full.py::test_from_summaries"

     

       
79
       
-
       
    "=tests/distributions/test_normal_full.py::test_serialization"

     

       
80
       
 
       
  ];

     

       
81
       
 
       


     

       
82
       
 
       
  disabledTests = [

     

···

       
3
       
 
       
  stdenv,

     

       
4
       
 
       
  buildPythonPackage,

     

       
5
       
 
       
  fetchFromGitHub,

     

       
6
       
+
       
  fetchpatch,

     

       
7
       
+
       
  pytestCheckHook,

     

       
8
       
 
       
  setuptools,

     

       
0
       
 
       

     

       
0
       
 
       

     

       
9
       
 
       
  apricot-select,

     

       
10
       
 
       
  networkx,

     

       
11
       
 
       
  numpy,

     

       
12
       
 
       
  scikit-learn,

     

       
13
       
 
       
  scipy,

     

       
14
       
 
       
  torch,

     

       
0
       
 
       

     

       
0
       
 
       

     

       
0
       
 
       

     

       
15
       
 
       
}:

     

       
16
       
 
       


     

       
17
       
 
       
buildPythonPackage rec {

     
···

       
22
       
 
       
  src = fetchFromGitHub {

     

       
23
       
 
       
    repo = "pomegranate";

     

       
24
       
 
       
    owner = "jmschrei";

     

       
25
       
+
       
    tag = "v${version}";

     

       
26
       
+
       
    hash = "sha256-p2Gn0FXnsAHvRUeAqx4M1KH0+XvDl3fmUZZ7MiMvPSs=";

     

       
0
       
 
       

     

       
0
       
 
       

     

       
27
       
 
       
  };

     

       
28
       
 
       


     

       
0
       
 
       

     

       
0
       
 
       

     

       
0
       
 
       

     

       
0
       
 
       

     

       
0
       
 
       

     

       
0
       
 
       

     

       
0
       
 
       

     

       
0
       
 
       

     

       
0
       
 
       

     

       
0
       
 
       

     

       
0
       
 
       

     

       
0
       
 
       

     

       
0
       
 
       

     

       
0
       
 
       

     

       
0
       
 
       

     

       
0
       
 
       

     

       
0
       
 
       

     

       
0
       
 
       

     

       
0
       
 
       

     

       
0
       
 
       

     

       
0
       
 
       

     

       
0
       
 
       

     

       
29
       
 
       
  build-system = [ setuptools ];

     

       
30
       
 
       


     

       
31
       
 
       
  dependencies = [

     
···

       
43
       
 
       
    pytestCheckHook

     

       
44
       
 
       
  ];

     

       
45
       
 
       


     

       
46
       
+
       
  patches = [

     

       
47
       
+
       
    # Fix tests for pytorch 2.6

     

       
48
       
+
       
    (fetchpatch {

     

       
49
       
+
       
      name = "python-2.6.patch";

     

       
50
       
+
       
      url = "https://github.com/jmschrei/pomegranate/pull/1142/commits/9ff5d5e2c959b44e569937e777b26184d1752a7b.patch";

     

       
51
       
+
       
      hash = "sha256-BXsVhkuL27QqK/n6Fa9oJCzrzNcL3EF6FblBeKXXSts=";

     

       
52
       
+
       
    })

     

       
53
       
+
       
  ];

     

       
54
       
+
       


     

       
55
       
+
       
  pytestFlagsArray = lib.optionals (stdenv.hostPlatform.isDarwin && stdenv.hostPlatform.isx86_64) [

     

       
56
       
 
       
    # AssertionError: Arrays are not almost equal to 6 decimals

     

       
57
       
+
       
    "--deselect=tests/distributions/test_normal_full.py::test_fit"

     

       
58
       
+
       
    "--deselect=tests/distributions/test_normal_full.py::test_from_summaries"

     

       
59
       
+
       
    "--deselect=tests/distributions/test_normal_full.py::test_serialization"

     

       
60
       
 
       
  ];

     

       
61
       
 
       


     

       
62
       
 
       
  disabledTests = [