Remove redundant jax.tree_util.register_dataclass(mbi.Dataset) call inside _checkpoint.py. - #238

Remove redundant jax.tree_util.register_dataclass(mbi.Dataset) call inside _checkpoint.py.#238

copybara-service[bot] merged 1 commit into

maingoogle/dpsynth:mainfrom

cl/995411189google/dpsynth:cl/995411189Copy head branch name to clipboard