commit 4c2f2f1a536e5850cff23ff800a4582307e30f7d · pyrox.dev/nixpkgs

+3

pkgs/development/python-modules/distrax/default.nix

···

       82
       82
        
           homepage = "https://github.com/deepmind/distrax";

     

       83
       83
        
           license = licenses.asl20;

     

       84
       84
        
           maintainers = with maintainers; [ onny ];

     

       85
       85
       +
           # Several tests fail with:

     

       86
       86
       +
           # AssertionError: [Chex] Assertion assert_type failed: Error in type compatibility check

     

       87
       87
       +
           broken = true;

     

       85
       88
        
         };

     

       86
       89
        
       }

+5

pkgs/development/python-modules/equinox/default.nix

···

       47
       47
        
       

     

       48
       48
        
         pythonImportsCheck = [ "equinox" ];

     

       49
       49
        
       

     

       50
       50
       +
         disabledTests = [

     

       51
       51
       +
           # Failed: DID NOT WARN. No warnings of type (<class 'UserWarning'>,) were emitted.

     

       52
       52
       +
           "test_tracetime"

     

       53
       53
       +
         ];

     

       54
       54
       +
       

     

       50
       55
        
         meta = with lib; {

     

       51
       56
        
           description = "A JAX library based around a simple idea: represent parameterised functions (such as neural networks) as PyTrees";

     

       52
       57
        
           changelog = "https://github.com/patrick-kidger/equinox/releases/tag/v${version}";

+4 -3

pkgs/development/python-modules/flax/default.nix

···

       25
       25
        
       

     

       26
       26
        
       buildPythonPackage rec {

     

       27
       27
        
         pname = "flax";

     

       28
       28
       -
         version = "0.8.1";

     

       28
       28
       +
         version = "0.8.2";

     

       29
       29
        
         pyproject = true;

     

       30
       30
        
       

     

       31
       31
       -
         disabled = pythonOlder "3.8";

     

       31
       31
       +
         disabled = pythonOlder "3.9";

     

       32
       32
        
       

     

       33
       33
        
         src = fetchFromGitHub {

     

       34
       34
        
           owner = "google";

     

       35
       35
        
           repo = "flax";

     

       36
       36
        
           rev = "refs/tags/v${version}";

     

       37
       37
       -
           hash = "sha256-3UzMSJoKw+V1WLBJ+Zf7aF7CDNBsvWnRUfNgb3K4v1A=";

     

       37
       37
       +
           hash = "sha256-UABgJGe1grUSkwOJpjeIoFqhXsqG//HlC1YyYPxXV+g=";

     

       38
       38
        
         };

     

       39
       39
        
       

     

       40
       40
        
         nativeBuildInputs = [

     
···

       87
       87
        
           # `tensorflow_datasets`, `vocabulary`) so the benefits of trying to run them

     

       88
       88
        
           # would be limited anyway.

     

       89
       89
        
           "examples/*"

     

       90
       90
       +
           "flax/experimental/nnx/examples/*"

     

       90
       91
        
           # See https://github.com/google/flax/issues/3232.

     

       91
       92
        
           "tests/jax_utils_test.py"

     

       92
       93
        
           # Requires tree

+10 -5

pkgs/development/python-modules/optax/default.nix

···

       1
       1
        
       { lib

     

       2
       2
        
       , absl-py

     

       3
       3
        
       , buildPythonPackage

     

       4
       4
       +
       , flit-core

     

       4
       5
        
       , chex

     

       5
       6
        
       , fetchFromGitHub

     

       6
       7
        
       , jaxlib

     
···

       11
       12
        
       

     

       12
       13
        
       buildPythonPackage rec {

     

       13
       14
        
         pname = "optax";

     

       14
       14
       -
         version = "0.2.1";

     

       15
       15
       -
         format = "setuptools";

     

       15
       15
       +
         version = "0.2.2";

     

       16
       16
       +
         pyproject = true;

     

       16
       17
        
       

     

       17
       17
       -
         disabled = pythonOlder "3.7";

     

       18
       18
       +
         disabled = pythonOlder "3.9";

     

       18
       19
        
       

     

       19
       20
        
         src = fetchFromGitHub {

     

       20
       21
        
           owner = "deepmind";

     

       21
       21
       -
           repo = pname;

     

       22
       22
       +
           repo = "optax";

     

       22
       23
        
           rev = "refs/tags/v${version}";

     

       23
       23
       -
           hash = "sha256-vimsVZV5Z11euLxsu998pMQZ0hG3xl96D3h9iONtl/E=";

     

       24
       24
       +
           hash = "sha256-sBiKUuQR89mttc9Njrh1aeUJOYdlcF7Nlj3/+Y7OMb4=";

     

       24
       25
        
         };

     

       25
       26
        
       

     

       26
       27
        
         outputs = [

     

       27
       28
        
           "out"

     

       28
       29
        
           "testsout"

     

       30
       30
       +
         ];

     

       31
       31
       +
       

     

       32
       32
       +
         nativeBuildInputs = [

     

       33
       33
       +
           flit-core

     

       29
       34
        
         ];

     

       30
       35
        
       

     

       31
       36
        
         buildInputs = [