Unverified Commit a7c18916 authored by Nick Cao's avatar Nick Cao Committed by GitHub
Browse files

python3Packages.orbax-checkpoint: 0.11.19 -> 0.11.20 (#429169)

parents d724c1b5 48039e02

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

+6 −4
Original line number Diff line number Diff line
@@ -9,6 +9,7 @@ 
  flit-core,



  # dependencies

  aiofiles,

  etils,

  humanize,

  importlib-resources,
@@ -23,7 +24,6 @@ 
  typing-extensions,



  # tests

  aiofiles,

  chex,

  google-cloud-logging,

  mock,
@@ -31,18 +31,19 @@ 
  portpicker,

  pytest-xdist,

  pytestCheckHook,

  safetensors,

}:



buildPythonPackage rec {

  pname = "orbax-checkpoint";

  version = "0.11.19";

  version = "0.11.20";

  pyproject = true;



  src = fetchFromGitHub {

    owner = "google";

    repo = "orbax";

    tag = "v${version}";

    hash = "sha256-j15E4jGvxIjEdWG6Lwr9mvPXr9WifrD1zFF6Vj+7wik=";

    hash = "sha256-ZZTmWW0PZHk8Evhk6JnbAyl7RQOAxN4GwXUv5M6GfIQ=";

  };



  sourceRoot = "${src.name}/checkpoint";
@@ -55,6 +56,7 @@ buildPythonPackage rec { 


  dependencies = [

    absl-py

    aiofiles

    etils

    humanize

    importlib-resources
@@ -70,7 +72,6 @@ buildPythonPackage rec { 
  ];



  nativeCheckInputs = [

    aiofiles

    chex

    google-cloud-logging

    mock
@@ -78,6 +79,7 @@ buildPythonPackage rec { 
    portpicker

    pytest-xdist

    pytestCheckHook

    safetensors

  ];



  pythonImportsCheck = [