Fix API reference docs to auto-discover all public modules

.gitignore

@@ -41,3 +41,4 @@ sagemaker_train/src/**/container_drivers/sm_train.sh
 sagemaker_train/src/**/container_drivers/sourcecode.json
 sagemaker_train/src/**/container_drivers/distributed.json
 .kiro
+docs/api/generated/

docs/api/sagemaker_core.rst

@@ -3,58 +3,8 @@ SageMaker Core
 
 Core SageMaker resources and utilities for managing AWS SageMaker services.
 
-.. currentmodule:: sagemaker.core
+.. autosummary::
+   :toctree: generated/
+   :recursive:
 
-Core Resources
---------------
-
-.. automodule:: sagemaker.core.resources
-   :members:
-   :undoc-members:
-   :show-inheritance:
-
-Session Management
-------------------
-
-.. automodule:: sagemaker.core.session_settings
-   :members:
-   :undoc-members:
-
-Configuration
--------------
-
-.. automodule:: sagemaker.core.config_schema
-   :members:
-   :undoc-members:
-
-Processing
-----------
-
-.. automodule:: sagemaker.core.processing
-   :members:
-   :undoc-members:
-
-Transformers
-------------
-
-.. automodule:: sagemaker.core.transformer
-   :members:
-   :undoc-members:
-
-Utilities
----------
-
-.. automodule:: sagemaker.core.common_utils
-   :members:
-   :undoc-members:
-
-.. automodule:: sagemaker.core.image_uris
-   :members:
-   :undoc-members:
-
-Exceptions
-----------
-
-.. automodule:: sagemaker.core.exceptions
-   :members:
-   :undoc-members:
+   sagemaker.core

docs/api/sagemaker_mlops.rst

@@ -3,37 +3,8 @@ SageMaker MLOps
 
 MLOps capabilities including pipelines, workflows, and model management.
 
-.. currentmodule:: sagemaker.mlops
+.. autosummary::
+   :toctree: generated/
+   :recursive:
 
-Pipeline Management
--------------------
-
-.. automodule:: sagemaker.mlops
-   :members:
-   :undoc-members:
-   :show-inheritance:
-
-Workflow Management
--------------------
-
-.. automodule:: sagemaker.mlops.workflow
-   :members:
-   :undoc-members:
-   :show-inheritance:
-
-Local Development
------------------
-
-.. automodule:: sagemaker.mlops.local
-   :members:
-   :undoc-members:
-   :show-inheritance:
-
-
-Feature Store
--------------
-
-.. automodule:: sagemaker.mlops.feature_store
-   :members:
-   :undoc-members:
-   :show-inheritance:
+   sagemaker.mlops

docs/api/sagemaker_serve.rst

@@ -3,12 +3,8 @@ SageMaker Serve
 
 Model serving and inference capabilities for deploying and managing ML models.
 
-.. currentmodule:: sagemaker.serve
+.. autosummary::
+   :toctree: generated/
+   :recursive:
 
-Model Deployment
-----------------
-
-.. automodule:: sagemaker.serve
-   :members:
-   :undoc-members:
-   :show-inheritance:
+   sagemaker.serve

docs/api/sagemaker_train.rst

@@ -1,30 +1,10 @@
 SageMaker Train
 ===============
 
-Training capabilities including model training, hyperparameter tuning, and distributed training.
+Training capabilities including model training, fine-tuning, hyperparameter tuning, and distributed training.
 
-.. currentmodule:: sagemaker.train
+.. autosummary::
+   :toctree: generated/
+   :recursive:
 
-Model Training
---------------
-
-.. automodule:: sagemaker.train
-   :members:
-   :undoc-members:
-   :show-inheritance:
-
-Distributed Training
---------------------
-
-.. automodule:: sagemaker.train.distributed
-   :members:
-   :undoc-members:
-   :show-inheritance:
-
-Model Evaluation
-----------------
-
-.. automodule:: sagemaker.train.evaluate
-   :members:
-   :undoc-members:
-   :show-inheritance:
+   sagemaker.train

docs/conf.py

@@ -39,6 +39,7 @@
 suppress_warnings = [
     'myst.header',  # Suppress header level warnings from notebooks
     'toc.not_readable',  # Suppress toctree warnings for symlinked files
+    'ref.python',  # Suppress "more than one target found" for duplicate class names across modules
 ]
 
 html_theme = 'sphinx_book_theme'
@@ -76,11 +77,28 @@
     'members': True,
     'undoc-members': True,
     'show-inheritance': True,
+    'private-members': False,
 }
 
-# Generate autosummary stubs
+# Generate autosummary stubs recursively
 autosummary_generate = True
 
+# Suppress internal/implementation modules not intended for users
+exclude_patterns += [
+    '*/telemetry*',
+    '*/tools*',
+    '*/container_drivers*',
+    '*/runtime_environment*',
+    '*/model_server*',
+    '*/detector*',
+    '*/validations*',
+]
+
+# Modules that fail to import due to runtime dependencies or side effects
+autodoc_mock_imports = [
+    'triton_python_backend_utils',
+]
+
 # Don't mock imports - let them fail gracefully and show what's available
 autodoc_mock_imports = []
 suppress_warnings = ['autodoc.import_error']