pyrox.dev
Merge pull request #291171 from fabaff/orbax-checkpoint
python311Packages.orbax-checkpoint: init at 0.5.3
authored by Fabian Affolter and committed by GitHub 9b640555 a54887d5
+78
pkgs/development/python-modules/orbax-checkpoint/default.nix
+78
pkgs/development/python-modules/orbax-checkpoint/default.nix
···
···
1
+
{ lib
2
+
, absl-py
3
+
, buildPythonPackage
4
+
, cached-property
5
+
, etils
6
+
, fetchPypi
7
+
, flit-core
8
+
, importlib-resources
9
+
, jax
10
+
, jaxlib
11
+
, msgpack
12
+
, nest-asyncio
13
+
, numpy
14
+
, protobuf
15
+
, pytest-xdist
16
+
, pytestCheckHook
17
+
, pythonOlder
18
+
, pyyaml
19
+
, tensorstore
20
+
, typing-extensions
21
+
}:
22
+
23
+
buildPythonPackage rec {
24
+
pname = "orbax-checkpoint";
25
+
version = "0.5.3";
26
+
pyproject = true;
27
+
28
+
disabled = pythonOlder "3.9";
29
+
30
+
src = fetchPypi {
31
+
pname = "orbax_checkpoint";
32
+
inherit version;
33
+
hash = "sha256-FXKQTLv+hROSfg2A+AtzDg7y9oAzLTwoENhENTKTi0U=";
34
+
};
35
+
36
+
nativeBuildInputs = [
37
+
flit-core
38
+
];
39
+
40
+
propagatedBuildInputs = [
41
+
absl-py
42
+
cached-property
43
+
etils
44
+
importlib-resources
45
+
jax
46
+
jaxlib
47
+
msgpack
48
+
nest-asyncio
49
+
numpy
50
+
protobuf
51
+
pyyaml
52
+
tensorstore
53
+
typing-extensions
54
+
];
55
+
56
+
nativeCheckInputs = [
57
+
pytest-xdist
58
+
pytestCheckHook
59
+
];
60
+
61
+
pythonImportsCheck = [
62
+
"orbax"
63
+
];
64
+
65
+
disabledTestPaths = [
66
+
# Circular dependency flax
67
+
"orbax/checkpoint/transform_utils_test.py"
68
+
"orbax/checkpoint/utils_test.py"
69
+
];
70
+
71
+
meta = with lib; {
72
+
description = "Orbax provides common utility libraries for JAX users";
73
+
homepage = "https://github.com/google/orbax/tree/main/checkpoint";
74
+
changelog = "https://github.com/google/orbax/blob/${version}/CHANGELOG.md";
75
+
license = licenses.asl20;
76
+
maintainers = with maintainers; [fab ];
77
+
};
78
+
}
+2
pkgs/top-level/python-packages.nix
+2
pkgs/top-level/python-packages.nix
···
8952
8953
oras = callPackage ../development/python-modules/oras { };
8954
8955
+
orbax-checkpoint = callPackage ../development/python-modules/orbax-checkpoint { };
8956
+
8957
orderedmultidict = callPackage ../development/python-modules/orderedmultidict { };
8958
8959
ordered-set = callPackage ../development/python-modules/ordered-set { };