Commit f222bbf8 authored by Gaetan Lepage's avatar Gaetan Lepage
Browse files

python3Packages.orbax-checkpoint: 0.11.13 -> 0.11.14 

Diff: https://github.com/google/orbax/compare/refs/tags/v0.11.13...refs/tags/v0.11.14

Changelog: https://github.com/google/orbax/blob/v0.11.14/checkpoint/CHANGELOG.md
parent cc3c0c7a

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

+16 −8
Original line number Diff line number Diff line
@@ -35,14 +35,14 @@ 


buildPythonPackage rec {

  pname = "orbax-checkpoint";

  version = "0.11.13";

  version = "0.11.14";

  pyproject = true;



  src = fetchFromGitHub {

    owner = "google";

    repo = "orbax";

    tag = "v${version}";

    hash = "sha256-qmq0Kz8wXUFFE4CqsdFwKXAIvysFbv7JomQSrNj1QCc=";

    hash = "sha256-qZfC3rqfESfXdL/TMXodrJnM1/dQs9adDHM9DS0QlZ4=";

  };



  sourceRoot = "${src.name}/checkpoint";
@@ -85,7 +85,14 @@ buildPythonPackage rec { 
    "orbax.checkpoint"

  ];



  disabledTests = lib.optionals stdenv.hostPlatform.isDarwin [

  disabledTests =

    [

      # Flaky

      # AssertionError: 2 not greater than 2.0046136379241943

      "test_async_mkdir_parallel"

      "test_async_mkdir_sequential"

    ]

    ++ lib.optionals stdenv.hostPlatform.isDarwin [

      # Probably failing because of a filesystem impurity

      # self.assertFalse(os.path.exists(dst_dir))

      # AssertionError: True is not false
@@ -105,6 +112,7 @@ buildPythonPackage rec { 
    "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/structure_utils_test.py"

    "orbax/checkpoint/_src/tree/utils_test.py"

    "orbax/checkpoint/single_host_test.py"

    "orbax/checkpoint/transform_utils_test.py"