Something went wrong. Try again.
What happens if I push all of nixpkgs
Something went wrong. Try again.
1.4 kB · 78 lines
Nix
at llvm-cleanup
12345678910111213141516171819202122232425262728293031323334353637383940414243444546474849505152535455565758596061626364656667686970717273747576777879{ lib, absl-py, buildPythonPackage, cached-property, etils, fetchPypi, flit-core, importlib-resources, jax, jaxlib, msgpack, nest-asyncio, numpy, protobuf, pytest-xdist, pytestCheckHook, pythonOlder, pyyaml, tensorstore, typing-extensions,}:
buildPythonPackage rec { pname = "orbax-checkpoint"; version = "0.5.9"; pyproject = true;
disabled = pythonOlder "3.9";
src = fetchPypi { pname = "orbax_checkpoint"; inherit version; hash = "sha256-H96IkUM3IxV79uddNBCU0dq+0dvPx8/Ps4HeCItGi2A="; };
build-system = [ flit-core ];
dependencies = [ absl-py cached-property etils importlib-resources jax jaxlib msgpack nest-asyncio numpy protobuf pyyaml tensorstore typing-extensions ];
nativeCheckInputs = [ pytest-xdist pytestCheckHook ];
pythonImportsCheck = [ "orbax" "orbax.checkpoint" ];
disabledTestPaths = [ # Circular dependency flax "orbax/checkpoint/transform_utils_test.py" "orbax/checkpoint/utils_test.py" ];
meta = with lib; { 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/${version}/CHANGELOG.md"; license = licenses.asl20; maintainers = with maintainers; [ fab ]; };}