Unverified Commit ca5fdb98 authored by Sandro Jäckel's avatar Sandro Jäckel Committed by GitHub
Browse files

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

parents 3daf7e2b 60abb755

pkgs/development/python-modules/orbax-checkpoint/default.nix

+10 −2
Original line number Diff line number Diff line
@@ -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"