Remove duplicate code

sagemaker-core/src/sagemaker/core/helper/session_helper.py

@@ -1669,22 +1669,10 @@ def _create_model_request(
         container_defs,
         vpc_config=None,
         enable_network_isolation=False,
-        primary_container=None,
         tags=None,
     ):  # pylint: disable=redefined-outer-name
         """Placeholder docstring"""
 
-        if container_defs and primary_container:
-            raise ValueError("Both container_defs and primary_container can not be passed as input")
-
-        if primary_container:
-            msg = (
-                "primary_container is going to be deprecated in a future release. Please use "
-                "container_defs instead."
-            )
-            warnings.warn(msg, DeprecationWarning)
-            container_defs = primary_container
-
         role = self.expand_role(role)
 
         if isinstance(container_defs, list):
@@ -1726,7 +1714,6 @@ def create_model(
         container_defs=None,
         vpc_config=None,
         enable_network_isolation=None,
-        primary_container=None,
         tags=None,
     ):
         """Create an Amazon SageMaker ``Model``.
@@ -1754,11 +1741,6 @@ def create_model(
                 * 'Subnets' (list[str]): List of subnet ids.
                 * 'SecurityGroupIds' (list[str]): List of security group ids.
             enable_network_isolation (bool): Whether the model requires network isolation or not.
-            primary_container (str or dict[str, str]): Docker image which defines the inference
-                code. You can also specify the return value of ``sagemaker.container_def()``,
-                which is used to create more advanced container configurations, including model
-                containers which need artifacts from S3. This field is deprecated, please use
-                container_defs instead.
             tags(Optional[Tags]): Optional. The list of tags to add to the model.
 
         Example:
@@ -1794,7 +1776,6 @@ def create_model(
             container_defs=container_defs,
             vpc_config=vpc_config,
             enable_network_isolation=enable_network_isolation,
-            primary_container=primary_container,
             tags=tags,
         )
 

sagemaker-core/src/sagemaker/core/lineage/action.py

@@ -38,7 +38,7 @@ class Action(_base_types.Record):
     Examples:
         .. code-block:: python
 
-            from sagemaker.lineage import action
+            from sagemaker.core.lineage import action
 
             my_action = action.Action.create(
                 action_name='MyAction',

sagemaker-core/src/sagemaker/core/lineage/artifact.py

@@ -42,7 +42,7 @@ class Artifact(_base_types.Record):
     Examples:
         .. code-block:: python
 
-            from sagemaker.lineage import artifact
+            from sagemaker.core.lineage import artifact
 
             my_artifact = artifact.Artifact.create(
                 artifact_name='MyArtifact',

sagemaker-core/src/sagemaker/core/lineage/association.py

@@ -31,7 +31,7 @@ class Association(_base_types.Record):
     Examples:
         .. code-block:: python
 
-            from sagemaker.lineage import association
+            from sagemaker.core.lineage import association
 
             my_association = association.Association.create(
                 source_arn=artifact_arn,

sagemaker-core/src/sagemaker/core/lineage/context.py

@@ -136,7 +136,7 @@ def load(cls, context_name: str, sagemaker_session=None) -> "Context":
         Examples:
             .. code-block:: python
 
-                from sagemaker.lineage import context
+                from sagemaker.core.lineage import context
 
                 my_context = context.Context.create(
                     context_name='MyContext',

sagemaker-core/src/sagemaker/core/lineage/query.py

@@ -194,9 +194,9 @@ def to_lineage_object(self):
             A ``Vertex`` object to its corresponding ``Artifact``,``Action``, ``Context``
             or ``TrialComponent`` object.
         """
-        from sagemaker.lineage.context import Context, EndpointContext
-        from sagemaker.lineage.action import Action
-        from sagemaker.lineage.lineage_trial_component import LineageTrialComponent
+        from sagemaker.core.lineage.context import Context, EndpointContext
+        from sagemaker.core.lineage.action import Action
+        from sagemaker.core.lineage.lineage_trial_component import LineageTrialComponent
 
         if self.lineage_entity == LineageEntityEnum.CONTEXT.value:
             resource_name = get_resource_name_from_arn(self.arn)
@@ -221,8 +221,8 @@ def to_lineage_object(self):
 
     def _artifact_to_lineage_object(self):
         """Convert the ``Vertex`` object to its corresponding ``Artifact``."""
-        from sagemaker.lineage.artifact import Artifact, ModelArtifact, ImageArtifact
-        from sagemaker.lineage.artifact import DatasetArtifact
+        from sagemaker.core.lineage.artifact import Artifact, ModelArtifact, ImageArtifact
+        from sagemaker.core.lineage.artifact import DatasetArtifact
 
         if self.lineage_source == LineageSourceEnum.MODEL.value:
             return ModelArtifact.load(artifact_arn=self.arn, sagemaker_session=self._session)

sagemaker-core/src/sagemaker/core/modules/distributed.py

@@ -19,8 +19,8 @@
 from typing import Optional, Dict, Any, List
 
 from sagemaker.core.modules.utils import safe_serialize
-from sagemaker.core.training.configs import BaseConfig
-from sagemaker.core.training.constants import SM_DRIVERS_LOCAL_PATH
+from sagemaker.train.configs import BaseConfig
+from sagemaker.train.constants import SM_DRIVERS_LOCAL_PATH
 
 
 class SMP(BaseConfig):