diff --git a/pkgs/development/python-modules/orbax-checkpoint/default.nix b/pkgs/development/python-modules/orbax-checkpoint/default.nix index 383a04d9a802..9ede2980a7c9 100644 --- a/pkgs/development/python-modules/orbax-checkpoint/default.nix +++ b/pkgs/development/python-modules/orbax-checkpoint/default.nix @@ -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"