Unverified Commit b45ff4af authored by OTABI Tomoya's avatar OTABI Tomoya Committed by GitHub
Browse files

Merge pull request #314651 from GaetanLepage/flax 

python311Packages.flax: 0.8.3 -> 0.8.4
parents 198bffc2 5d96b117

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

+6 −6
Original line number Diff line number Diff line
@@ -26,7 +26,7 @@ 


buildPythonPackage rec {

  pname = "flax";

  version = "0.8.3";

  version = "0.8.4";

  pyproject = true;



  disabled = pythonOlder "3.9";
@@ -35,7 +35,7 @@ buildPythonPackage rec { 
    owner = "google";

    repo = "flax";

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

    hash = "sha256-uDGTyksUZTTL6FiTJP+qteFLOjr75dcTj9yRJ6Jm8xU=";

    hash = "sha256-ZwqKZdJ9LOfWTav5nE9xMsMw/DbryqQUuu5fqeugBzY=";

  };



  build-system = [
@@ -86,7 +86,7 @@ buildPythonPackage rec { 
    # `tensorflow_datasets`, `vocabulary`) so the benefits of trying to run them

    # would be limited anyway.

    "examples/*"

    "flax/experimental/nnx/examples/*"

    "flax/nnx/examples/*"

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

    "tests/jax_utils_test.py"

    # Requires tree
@@ -98,11 +98,11 @@ buildPythonPackage rec { 
    "test_overwrite_checkpoints0"

  ];



  meta = with lib; {

  meta = {

    description = "Neural network library for JAX";

    homepage = "https://github.com/google/flax";

    changelog = "https://github.com/google/flax/releases/tag/v${version}";

    license = licenses.asl20;

    maintainers = with maintainers; [ ndl ];

    license = lib.licenses.asl20;

    maintainers = with lib.maintainers; [ ndl ];

  };

}