python312Packages.orbax-checkpoint: 0.11.0 -> 0.11.2 (NixOS#377641)

Author: SuperSandro2000

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"