Fix readthedoc build issue

PiperOrigin-RevId: 695407901
google · Nov 11, 2024 · c2d6863 · c2d6863
1 parent 9e18554
commit c2d6863
Show file tree

Hide file tree

Showing 3 changed files with 9 additions and 2 deletions.
diff --git a/checkpoint/CHANGELOG.md b/checkpoint/CHANGELOG.md
@@ -7,9 +7,14 @@ and this project adheres to [Semantic Versioning](https://semver.org/spec/v2.0.0
 
 ## [Unreleased]
 
+## [0.9.1] - 2024-11-11
 ### Added
 - Introduce `CheckpointManagerOptions.should_keep_fn` as an alternative to Introduce `CheckpointManagerOptions.keep_period`.
 
+### Fixed
+
+- Fix readthedoc build failures on source files in `_src`
+
 ## [0.9.0] - 2024-11-08
 ### Changed
 - Create `Composite` class, which `CompositeArgs` now subclasses.

diff --git a/checkpoint/orbax/checkpoint/version.py b/checkpoint/orbax/checkpoint/version.py
@@ -16,7 +16,7 @@
 
 # A new PyPI release will be pushed everytime `__version__` is increased.
 # Also modify version and date in CHANGELOG.
-__version__ = '0.9.0'
+__version__ = '0.9.1'
 
 
 # TODO: b/362813406 - Add latest change timestamp and commit number.

diff --git a/docs/conf.py b/docs/conf.py
@@ -133,7 +133,9 @@ def linkcode_resolve(domain, info):
 
   try:
     mod = sys.modules[info['module']]
-  except ImportError:
+  except (ImportError, KeyError):
+    # TODO(dnlng) - Add support for resolving source code links for
+    # modules that are in _src
     return None
 
   obj = mod