commit 7455df3c51e07a99474c6751a4433c0a5046110b · pyrox.dev/nixpkgs

+4 -7
pkgs/development/python-modules/numpyro/default.nix
···

       16
       16
        
       

     

       17
       17
        
         # tests

     

       18
       18
        
         dm-haiku,

     

       19
       19
       +
         equinox,

     

       19
       20
        
         flax,

     

       20
       21
        
         funsor,

     

       21
       22
        
         graphviz,

     
···

       29
       30
        
       

     

       30
       31
        
       buildPythonPackage rec {

     

       31
       32
        
         pname = "numpyro";

     

       32
       32
       -
         version = "0.18.0";

     

       33
       33
       +
         version = "0.19.0";

     

       33
       34
        
         pyproject = true;

     

       34
       35
        
       

     

       35
       36
        
         src = fetchFromGitHub {

     

       36
       37
        
           owner = "pyro-ppl";

     

       37
       38
        
           repo = "numpyro";

     

       38
       39
        
           tag = version;

     

       39
       39
       -
           hash = "sha256-0X/ta2yfzjf3JnZYdUAzQmXvbsDpwFCJe/bArMSWQgU=";

     

       40
       40
       +
           hash = "sha256-3kzaINsz1Mjk97ERQsQIYIBz7CVmXtVDn0edJFMHQWs=";

     

       40
       41
        
         };

     

       41
       42
        
       

     

       42
       43
        
         build-system = [ setuptools ];

     
···

       51
       52
        
       

     

       52
       53
        
         nativeCheckInputs = [

     

       53
       54
        
           dm-haiku

     

       55
       55
       +
           equinox

     

       54
       56
        
           flax

     

       55
       57
        
           funsor

     

       56
       58
        
           graphviz

     
···

       92
       94
        
       

     

       93
       95
        
           # ValueError: compiling computation that requires 2 logical devices, but only 1 XLA devices are available (num_replicas=2)

     

       94
       96
        
           "test_chain"

     

       95
       95
       -
       

     

       96
       96
       -
           # Failing since flax==0.11.0

     

       97
       97
       -
           # KeyError: "No RngStream named 'dropout' found in Rngs."

     

       98
       98
       -
           # https://github.com/pyro-ppl/numpyro/issues/2055

     

       99
       99
       -
           "test_nnx_state_dropout_smoke"

     

       100
       97
        
         ]

     

       101
       98
        
         ++ lib.optionals stdenv.hostPlatform.isDarwin [

     

       102
       99
        
           # AssertionError: Not equal to tolerance rtol=0.06, atol=0