Increase Orbax version.

PiperOrigin-RevId: 520469794
google · Mar 29, 2023 · cde8f95 · cde8f95
1 parent 2d041de
commit cde8f95
Show file tree

Hide file tree

Showing 3 changed files with 4 additions and 1 deletion.
diff --git a/CHANGELOG.md b/CHANGELOG.md
@@ -7,6 +7,8 @@ and this project adheres to [Semantic Versioning](https://semver.org/spec/v2.0.0
 
 ## [Unreleased]
 
+## [0.1.7] - 2022-03-29
+
 ### Added
 - Support for OCDBT driver in Tensorstore.
 

diff --git a/orbax/__init__.py b/orbax/__init__.py
@@ -15,4 +15,4 @@
 """Orbax API."""
 
 # A new PyPI release will be pushed everytime `__version__` is increased.
-__version__ = '0.1.6'
+__version__ = '0.1.7'
diff --git a/orbax/checkpoint/__init__.py b/orbax/checkpoint/__init__.py
@@ -19,6 +19,7 @@
 from orbax.checkpoint import aggregate_handlers
 from orbax.checkpoint import checkpoint_utils
 from orbax.checkpoint import lazy_utils
+from orbax.checkpoint import msgpack_utils
 from orbax.checkpoint import test_utils
 from orbax.checkpoint import type_handlers
 from orbax.checkpoint import utils