commit f2abb2fbc865094f63b1fed5e30003bacd6ce60c · pyrox.dev/nixpkgs

+17 -37
pkgs/development/python-modules/pomegranate/default.nix
···

       3
       3
        
         stdenv,

     

       4
       4
        
         buildPythonPackage,

     

       5
       5
        
         fetchFromGitHub,

     

       6
       6
       -
       

     

       7
       7
       -
         # build-system

     

       6
       6
       +
         fetchpatch,

     

       7
       7
       +
         pytestCheckHook,

     

       8
       8
        
         setuptools,

     

       9
       9
       -
       

     

       10
       10
       -
         # dependencies

     

       11
       9
        
         apricot-select,

     

       12
       10
        
         networkx,

     

       13
       11
        
         numpy,

     

       14
       12
        
         scikit-learn,

     

       15
       13
        
         scipy,

     

       16
       14
        
         torch,

     

       17
       17
       -
       

     

       18
       18
       -
         # tests

     

       19
       19
       -
         pytestCheckHook,

     

       20
       15
        
       }:

     

       21
       16
        
       

     

       22
       17
        
       buildPythonPackage rec {

     
···

       27
       22
        
         src = fetchFromGitHub {

     

       28
       23
        
           repo = "pomegranate";

     

       29
       24
        
           owner = "jmschrei";

     

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

     

       31
       31
       -
           # No tag for 1.1.2

     

       32
       32
       -
           rev = "e9162731f4f109b7b17ecffde768734cacdb839b";

     

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

     

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

     

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

     

       34
       27
        
         };

     

       35
       28
        
       

     

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

     

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

     

       38
       38
       -
         postPatch = ''

     

       39
       39
       -
           substituteInPlace \

     

       40
       40
       -
             tests/distributions/test_bernoulli.py \

     

       41
       41
       -
             tests/distributions/test_categorical.py \

     

       42
       42
       -
             tests/distributions/test_exponential.py \

     

       43
       43
       -
             tests/distributions/test_gamma.py \

     

       44
       44
       -
             tests/distributions/test_independent_component.py \

     

       45
       45
       -
             tests/distributions/test_normal_diagonal.py \

     

       46
       46
       -
             tests/distributions/test_normal_full.py \

     

       47
       47
       -
             tests/distributions/test_poisson.py \

     

       48
       48
       -
             tests/distributions/test_student_t.py \

     

       49
       49
       -
             tests/distributions/test_uniform.py \

     

       50
       50
       -
             tests/test_bayes_classifier.py \

     

       51
       51
       -
             tests/test_gmm.py \

     

       52
       52
       -
             tests/test_kmeans.py \

     

       53
       53
       -
             --replace-fail \

     

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

     

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

     

       56
       56
       -
         '';

     

       57
       57
       -
       

     

       58
       29
        
         build-system = [ setuptools ];

     

       59
       30
        
       

     

       60
       31
        
         dependencies = [

     
···

       72
       43
        
           pytestCheckHook

     

       73
       44
        
         ];

     

       74
       45
        
       

     

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

     

       46
       46
       +
         patches = [

     

       47
       47
       +
           # Fix tests for pytorch 2.6

     

       48
       48
       +
           (fetchpatch {

     

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

     

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

     

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

     

       52
       52
       +
           })

     

       53
       53
       +
         ];

     

       54
       54
       +
       

     

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

     

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

     

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

     

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

     

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

     

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

     

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

     

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

     

       80
       60
        
         ];

     

       81
       61
        
       

     

       82
       62
        
         disabledTests = [