From 6eaf3b1dc478c4e4e963ce81dfed2f04010153bf Mon Sep 17 00:00:00 2001
From: Harshal Sheth <hsheth2@gmail.com>
Date: Thu, 15 Dec 2022 18:31:12 -0500
Subject: [PATCH] docs(ingest): fix error in custom tags transformer example
 (#6767)

---
 .../docs/transformer/dataset_transformer.md     | 17 +++--------------
 1 file changed, 3 insertions(+), 14 deletions(-)

diff --git a/metadata-ingestion/docs/transformer/dataset_transformer.md b/metadata-ingestion/docs/transformer/dataset_transformer.md
index 8478664f31b00..753e9c87300d9 100644
--- a/metadata-ingestion/docs/transformer/dataset_transformer.md
+++ b/metadata-ingestion/docs/transformer/dataset_transformer.md
@@ -309,22 +309,11 @@ import logging
 
 import datahub.emitter.mce_builder as builder
 from datahub.metadata.schema_classes import (
-    DatasetSnapshotClass,
     TagAssociationClass
 )
 
-def custom_tags(current: DatasetSnapshotClass) -> List[TagAssociationClass]:
-    """ Returns tags to associate to a dataset depending on custom logic
-
-    This function receives a DatasetSnapshotClass, performs custom logic and returns
-    a list of TagAssociationClass-wrapped tags.
-
-    Args:
-        current (DatasetSnapshotClass): Single DatasetSnapshotClass object
-
-    Returns:
-        List of TagAssociationClass objects.
-    """
+def custom_tags(entity_urn: str) -> List[TagAssociationClass]:
+    """Compute the tags to associate to a given dataset."""
 
     tag_strings = []
 
@@ -335,7 +324,7 @@ def custom_tags(current: DatasetSnapshotClass) -> List[TagAssociationClass]:
     tag_strings = [builder.make_tag_urn(tag=n) for n in tag_strings]
     tags = [TagAssociationClass(tag=tag) for tag in tag_strings]
     
-    logging.info(f"Tagging dataset {current.urn} with {tag_strings}.")
+    logging.info(f"Tagging dataset {entity_urn} with {tag_strings}.")
     return tags
 ```
 Finally, you can install and use your custom transformer as [shown here](#installing-the-package).