Adjust module directives in checkpoint_manager.rst to use simpler namespaces.

Orbax Authors · Orbax Authors · commit 17b4717764d4 · 2025-08-13T08:53:45.000-07:00
PiperOrigin-RevId: 794601067
diff --git a/docs/api_reference/checkpoint.checkpoint_manager.rst b/docs/api_reference/checkpoint.checkpoint_manager.rst
@@ -1,10 +1,10 @@
 CheckpointManager
 ============================================================================
 
-.. currentmodule:: orbax.checkpoint.checkpoint_manager
-
 .. automodule:: orbax.checkpoint.checkpoint_manager
 
+.. currentmodule:: orbax.checkpoint
+
 CheckpointManager
 ------------------------------------------------
 .. autoclass:: CheckpointManager
@@ -15,17 +15,8 @@ CheckpointManagerOptions
 .. autoclass:: CheckpointManagerOptions
   :members:
 
-AsyncOptions
-------------------------------------------------
-.. autoclass:: AsyncOptions
-  :members:
-
-MultiprocessingOptions
-------------------------------------------------
-.. autoclass:: MultiprocessingOptions
-  :members:
-
 Utility functions
 ------------------------------------------------
+.. currentmodule:: orbax.checkpoint.checkpoint_manager
 
 .. autofunction:: is_async_checkpointer
