Merge pull request #314651 from GaetanLepage/flax

python311Packages.flax: 0.8.3 -> 0.8.4
This commit is contained in:
OTABI Tomoya 2024-05-26 08:54:09 +09:00 committed by GitHub
commit b45ff4af4f
No known key found for this signature in database
GPG Key ID: B5690EEEBB952194

View File

@ -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 ];
};
}