Gaetan Lepage
2025-05-01 23:23:23 +02:00
parent f149b2b05a
commit 9750c0094b
@@ -35,14 +35,14 @@
buildPythonPackage rec {
pname = "orbax-checkpoint";
version = "0.11.12";
version = "0.11.13";
pyproject = true;
src = fetchFromGitHub {
owner = "google";
repo = "orbax";
tag = "v${version}";
hash = "sha256-pwp3YIRiR17TBDu7ILmf0pi37biQ2hr7QtWKqBPVP2A=";
hash = "sha256-qmq0Kz8wXUFFE4CqsdFwKXAIvysFbv7JomQSrNj1QCc=";
};
sourceRoot = "${src.name}/checkpoint";
@@ -104,6 +104,7 @@ buildPythonPackage rec {
"orbax/checkpoint/_src/metadata/tree_rich_types_test.py"
"orbax/checkpoint/_src/metadata/tree_test.py"
"orbax/checkpoint/_src/testing/test_tree_utils.py"
"orbax/checkpoint/_src/tree/parts_of_test.py"
"orbax/checkpoint/_src/tree/utils_test.py"
"orbax/checkpoint/single_host_test.py"
"orbax/checkpoint/transform_utils_test.py"