package(default_visibility = ["//visibility:public"])

py_library(
    name = "types",
    srcs = ["types.py"],
    deps = ["//checkpoint/orbax/checkpoint/_src/tree:types"],
)
