Unverified Commit 44f7b5e8 authored by Gaétan Lepage's avatar Gaétan Lepage Committed by GitHub
Browse files

python3Packages.orbax-checkpoint: 0.11.30 -> 0.11.33 (#499410)

parents 65b8a0dc 4c245cda

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

+14 −10
Original line number Diff line number Diff line
@@ -14,7 +14,6 @@ 
  humanize,

  jax,

  msgpack,

  nest-asyncio,

  numpy,

  protobuf,

  psutil,
@@ -22,10 +21,13 @@ 
  simplejson,

  tensorstore,

  typing-extensions,

  uvloop,



  # tests

  chex,

  fastapi,

  google-cloud-logging,

  httpx,

  mock,

  optax,

  portpicker,
@@ -35,19 +37,19 @@ 
  torch,

}:



buildPythonPackage rec {

buildPythonPackage (finalAttrs: {

  pname = "orbax-checkpoint";

  version = "0.11.30";

  version = "0.11.33";

  pyproject = true;



  src = fetchFromGitHub {

    owner = "google";

    repo = "orbax";

    tag = "v${version}";

    hash = "sha256-y8l0AVGt2t5zLX+x+yuWHsEDy68agpXIkrew+zfYGXU=";

    tag = "v${finalAttrs.version}";

    hash = "sha256-ibHV+MwQvlh2USeDVAUWEJxCS1MGuPZRnZZTBWFO7UQ=";

  };



  sourceRoot = "${src.name}/checkpoint";

  sourceRoot = "${finalAttrs.src.name}/checkpoint";



  build-system = [ flit-core ];
@@ -58,7 +60,6 @@ buildPythonPackage rec { 
    humanize

    jax

    msgpack

    nest-asyncio

    numpy

    protobuf

    psutil
@@ -66,13 +67,16 @@ buildPythonPackage rec { 
    simplejson

    tensorstore

    typing-extensions

    uvloop

  ]

  ++ etils.optional-dependencies.epath

  ++ etils.optional-dependencies.epy;



  nativeCheckInputs = [

    chex

    fastapi

    google-cloud-logging

    httpx

    mock

    optax

    portpicker
@@ -81,7 +85,6 @@ buildPythonPackage rec { 
    safetensors

    torch

  ];



  pythonImportsCheck = [

    "orbax"

    "orbax.checkpoint"
@@ -156,13 +159,14 @@ buildPythonPackage rec { 
    "orbax/checkpoint/checkpoint_manager_test.py"

    "orbax/checkpoint/single_host_test.py"

    "orbax/checkpoint/transform_utils_test.py"

    "orbax/checkpoint/_src/handlers/pytree_checkpoint_handler_test.py"

  ];



  meta = {

    description = "Orbax provides common utility libraries for JAX users";

    homepage = "https://github.com/google/orbax/tree/main/checkpoint";

    changelog = "https://github.com/google/orbax/blob/v${version}/checkpoint/CHANGELOG.md";

    changelog = "https://github.com/google/orbax/blob/v${finalAttrs.version}/checkpoint/CHANGELOG.md";

    license = lib.licenses.asl20;

    maintainers = with lib.maintainers; [ fab ];

  };

}
 
})