fix: deprecate tag logic on Association (#2552)

Co-authored-by: Shreya Pandit <[email protected]>

Author: danabens

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