diff --git a/src/sagemaker/lineage/association.py b/src/sagemaker/lineage/association.py index a0b1ce7e85..fef79e2f8f 100644 --- a/src/sagemaker/lineage/association.py +++ b/src/sagemaker/lineage/association.py @@ -15,11 +15,14 @@ from typing import Optional, Iterator from datetime import datetime +import logging from sagemaker.apiutils import _base_types from sagemaker.lineage import _api_types from sagemaker.lineage._api_types import AssociationSummary +logger = logging.getLogger(__name__) + class Association(_base_types.Record): """An Amazon SageMaker artifact, which is part of a SageMaker lineage. @@ -73,6 +76,10 @@ def set_tag(self, tag=None): Returns: list({str:str}): a list of key value pairs """ + logger.warning( + "set_tag on Association is deprecated. Use set_tag on the source or destination\ + entity instead." + ) return self._set_tags(resource_arn=self.source_arn, tags=[tag]) def set_tags(self, tags=None): @@ -84,6 +91,10 @@ def set_tags(self, tags=None): Returns: list({str:str}): a list of key value pairs """ + logger.warning( + "set_tags on Association is deprecated. Use set_tags on the source or destination\ + entity instead." + ) return self._set_tags(resource_arn=self.source_arn, tags=tags) @classmethod