python312Packages.orbax-checkpoint: 0.11.0 -> 0.11.2
Diff: https://github.com/google/orbax/compare/refs/tags/v0.11.0...v0.11.2 Changelog: https://github.com/google/orbax/blob/v0.11.2/checkpoint/CHANGELOG.md
This commit is contained in:
@@ -27,20 +27,21 @@
|
||||
google-cloud-logging,
|
||||
mock,
|
||||
optax,
|
||||
portpicker,
|
||||
pytest-xdist,
|
||||
pytestCheckHook,
|
||||
}:
|
||||
|
||||
buildPythonPackage rec {
|
||||
pname = "orbax-checkpoint";
|
||||
version = "0.11.0";
|
||||
version = "0.11.2";
|
||||
pyproject = true;
|
||||
|
||||
src = fetchFromGitHub {
|
||||
owner = "google";
|
||||
repo = "orbax";
|
||||
tag = "v${version}";
|
||||
hash = "sha256-pVRXWJfiiqV2ZFM0CgXdwD6/lnRa1HFFPrfS5975mVA=";
|
||||
hash = "sha256-68mqz+4FJQr/mKR53L5CSenRB0SjV3dbBdtBylFW1y8=";
|
||||
};
|
||||
|
||||
sourceRoot = "${src.name}/checkpoint";
|
||||
@@ -68,6 +69,7 @@ buildPythonPackage rec {
|
||||
google-cloud-logging
|
||||
mock
|
||||
optax
|
||||
portpicker
|
||||
pytest-xdist
|
||||
pytestCheckHook
|
||||
];
|
||||
@@ -85,6 +87,12 @@ buildPythonPackage rec {
|
||||
];
|
||||
|
||||
disabledTestPaths = [
|
||||
# E absl.flags._exceptions.DuplicateFlagError: The flag 'num_processes' is defined twice.
|
||||
# First from multiprocess_test, Second from orbax.checkpoint._src.testing.multiprocess_test.
|
||||
# Description from first occurrence: Number of processes to use.
|
||||
# https://github.com/google/orbax/issues/1580
|
||||
"orbax/checkpoint/experimental/emergency/"
|
||||
|
||||
# Circular dependency flax
|
||||
"orbax/checkpoint/_src/metadata/empty_values_test.py"
|
||||
"orbax/checkpoint/_src/metadata/tree_rich_types_test.py"
|
||||
|
||||
Reference in New Issue
Block a user