remove lazy loading and try model_builder.rst update

Author: mollyheamazon

doc/api/inference/model_builder.rst

@@ -3,14 +3,14 @@ Model Builder
 
 This module contains classes related to Amazon Sagemaker Model Builder
 
-.. autoclass:: sagemaker.serve.builder.model_builder.ModelBuilder
+.. autoclass:: sagemaker.serve.ModelBuilder
 
-.. automethod:: sagemaker.serve.builder.model_builder.ModelBuilder.build
+.. automethod:: sagemaker.serve.ModelBuilder.build
 
-.. automethod:: sagemaker.serve.builder.model_builder.ModelBuilder.save
+.. automethod:: sagemaker.serve.ModelBuilder.save
 
-.. autoclass:: sagemaker.serve.spec.inference_spec.InferenceSpec
+.. autoclass:: sagemaker.serve.InferenceSpec
 
-.. autoclass:: sagemaker.serve.builder.schema_builder.SchemaBuilder
+.. autoclass:: sagemaker.serve.SchemaBuilder
 
-.. autoclass:: sagemaker.serve.marshalling.custom_payload_translator.CustomPayloadTranslator
+.. autoclass:: sagemaker.serve.CustomPayloadTranslator

src/sagemaker/serve/__init__.py

@@ -3,12 +3,7 @@
 from __future__ import absolute_import
 import logging
 
-# from sagemaker.serve.builder.model_builder import ModelBuilder
-from typing import TYPE_CHECKING
-
-if TYPE_CHECKING:
-    from sagemaker.serve.builder.model_builder import ModelBuilder
-
+from sagemaker.serve.builder.model_builder import ModelBuilder
 from sagemaker.serve.utils.types import ModelServer
 from sagemaker.serve.spec.inference_spec import InferenceSpec
 from sagemaker.serve.mode.function_pointers import Mode