prompt: We are currently solving the following issue within our repository. Here is the issue text:
  --- BEGIN ISSUE ---
  [FEA] Simplify AddMetadata Tag for NVTabular
  **Is your feature request related to a problem? Please describe.**
  Currently, we provide the functionality for the user to tag columns with the operator `AddMetadata`.
  The use case is that users will use the operator mainly for adding tags. 
  
  Should we provide a wrapper called `AddTag` or `TagAs` to simplify the AddMetadata operator?
  Should we provide multiple wrappers for common tags - e.g.
  
  `TagAsUserID()`, `TagAsItemID()`, `TagAsUserFeatures()`, `TagAsItemFeatures()`, etc.
  
  
  
  --- END ISSUE ---
  
  Below are some code segments, each from a relevant file. One or more of these files may contain bugs.
  --- BEGIN FILES ---
  Path: `nvtabular/ops/add_metadata.py`
  Content:
  ```
  1 #
  2 # Copyright (c) 2021, NVIDIA CORPORATION.
  3 #
  4 # Licensed under the Apache License, Version 2.0 (the "License");
  5 # you may not use this file except in compliance with the License.
  6 # You may obtain a copy of the License at
  7 #
  8 #     http://www.apache.org/licenses/LICENSE-2.0
  9 #
  10 # Unless required by applicable law or agreed to in writing, software
  11 # distributed under the License is distributed on an "AS IS" BASIS,
  12 # WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
  13 # See the License for the specific language governing permissions and
  14 # limitations under the License.
  15 #
  16 from nvtabular.dispatch import DataFrameType
  17 
  18 from .operator import ColumnSelector, Operator
  19 
  20 
  21 class AddMetadata(Operator):
  22     """
  23     This operator will add user defined tags and properties
  24     to a Schema.
  25     """
  26 
  27     def __init__(self, tags=None, properties=None):
  28         super().__init__()
  29         self.tags = tags or []
  30         self.properties = properties or {}
  31 
  32     def transform(self, col_selector: ColumnSelector, df: DataFrameType) -> DataFrameType:
  33         return df
  34 
  35     @property
  36     def output_tags(self):
  37         return self.tags
  38 
  39     @property
  40     def output_properties(self):
  41         return self.properties
  42
  ```
  
  Path: `nvtabular/ops/__init__.py`
  Content:
  ```
  1 #
  2 # Copyright (c) 2021, NVIDIA CORPORATION.
  3 #
  4 # Licensed under the Apache License, Version 2.0 (the "License");
  5 # you may not use this file except in compliance with the License.
  6 # You may obtain a copy of the License at
  7 #
  8 #     http://www.apache.org/licenses/LICENSE-2.0
  9 #
  10 # Unless required by applicable law or agreed to in writing, software
  11 # distributed under the License is distributed on an "AS IS" BASIS,
  12 # WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
  13 # See the License for the specific language governing permissions and
  14 # limitations under the License.
  15 #
  16 
  17 # alias submodules here to avoid breaking everything with moving to submodules
  18 # flake8: noqa
  19 from .add_metadata import AddMetadata
  20 from .bucketize import Bucketize
  21 from .categorify import Categorify, get_embedding_sizes
  22 from .clip import Clip
  23 from .column_similarity import ColumnSimilarity
  24 from .data_stats import DataStats
  25 from .difference_lag import DifferenceLag
  26 from .drop_low_cardinality import DropLowCardinality
  27 from .dropna import Dropna
  28 from .fill import FillMedian, FillMissing
  29 from .filter import Filter
  30 from .groupby import Groupby
  31 from .hash_bucket import HashBucket
  32 from .hashed_cross import HashedCross
  33 from .join_external import JoinExternal
  34 from .join_groupby import JoinGroupby
  35 from .lambdaop import LambdaOp
  36 from .list_slice import ListSlice
  37 from .logop import LogOp
  38 from .normalize import Normalize, NormalizeMinMax
  39 from .operator import ColumnSelector, Operator
  40 from .reduce_dtype_size import ReduceDtypeSize
  41 from .rename import Rename
  42 from .stat_operator import StatOperator
  43 from .target_encoding import TargetEncoding
  44 from .value_counts import ValueCount
  45
  ```
  --- END FILES ---
  
  Please first localize the bug based on the issue statement, and then generate a patch according to the `git diff` format fenced by three backticks.
  
  Here is an example:
  
  ```diff
  diff --git a/examples/server_async.py b/examples/server_async.py
  --- a/examples/server_async.py
  +++ b/examples/server_async.py
  @@ -313,4 +313,4 @@
   
   
   if __name__ == "__main__":
  -    asyncio.run(run_async_server("."), debug=True)
  +    asyncio.run(run_async_server(), debug=True)
  diff --git a/examples/server_sync.py b/examples/server_sync.py
  --- a/examples/server_sync.py
  +++ b/examples/server_sync.py
  @@ -313,5 +313,5 @@
   
   
   if __name__ == "__main__":
  -    server = run_sync_server(".")
  +    server = run_sync_server()
       server.shutdown()
  
  ```
  
golden_diff: diff --git a/nvtabular/ops/__init__.py b/nvtabular/ops/__init__.py
  --- a/nvtabular/ops/__init__.py
  +++ b/nvtabular/ops/__init__.py
  @@ -16,7 +16,15 @@
   
   # alias submodules here to avoid breaking everything with moving to submodules
   # flake8: noqa
  -from .add_metadata import AddMetadata
  +from .add_metadata import (
  +    AddMetadata,
  +    AddProperties,
  +    AddTags,
  +    TagAsItemFeatures,
  +    TagAsItemID,
  +    TagAsUserFeatures,
  +    TagAsUserID,
  +)
   from .bucketize import Bucketize
   from .categorify import Categorify, get_embedding_sizes
   from .clip import Clip
  diff --git a/nvtabular/ops/add_metadata.py b/nvtabular/ops/add_metadata.py
  --- a/nvtabular/ops/add_metadata.py
  +++ b/nvtabular/ops/add_metadata.py
  @@ -13,6 +13,7 @@
   # See the License for the specific language governing permissions and
   # limitations under the License.
   #
  +from merlin.schema.tags import Tags
   from nvtabular.dispatch import DataFrameType
   
   from .operator import ColumnSelector, Operator
  @@ -39,3 +40,38 @@
       @property
       def output_properties(self):
           return self.properties
  +
  +
  +class AddTags(AddMetadata):
  +    def __init__(self, tags=None):
  +        super().__init__(tags=tags)
  +
  +
  +class AddProperties(AddMetadata):
  +    def __init__(self, properties=None):
  +        super().__init__(properties=properties)
  +
  +
  +# Wrappers for common features
  +class TagAsUserID(Operator):
  +    @property
  +    def output_tags(self):
  +        return [Tags.USER_ID]
  +
  +
  +class TagAsItemID(Operator):
  +    @property
  +    def output_tags(self):
  +        return [Tags.ITEM_ID]
  +
  +
  +class TagAsUserFeatures(Operator):
  +    @property
  +    def output_tags(self):
  +        return [Tags.USER]
  +
  +
  +class TagAsItemFeatures(Operator):
  +    @property
  +    def output_tags(self):
  +        return [Tags.ITEM]
  
verification_info: {"golden_diff": "diff --git a/nvtabular/ops/__init__.py b/nvtabular/ops/__init__.py\n--- a/nvtabular/ops/__init__.py\n+++ b/nvtabular/ops/__init__.py\n@@ -16,7 +16,15 @@\n \n # alias submodules here to avoid breaking everything with moving to submodules\n # flake8: noqa\n-from .add_metadata import AddMetadata\n+from .add_metadata import (\n+    AddMetadata,\n+    AddProperties,\n+    AddTags,\n+    TagAsItemFeatures,\n+    TagAsItemID,\n+    TagAsUserFeatures,\n+    TagAsUserID,\n+)\n from .bucketize import Bucketize\n from .categorify import Categorify, get_embedding_sizes\n from .clip import Clip\ndiff --git a/nvtabular/ops/add_metadata.py b/nvtabular/ops/add_metadata.py\n--- a/nvtabular/ops/add_metadata.py\n+++ b/nvtabular/ops/add_metadata.py\n@@ -13,6 +13,7 @@\n # See the License for the specific language governing permissions and\n # limitations under the License.\n #\n+from merlin.schema.tags import Tags\n from nvtabular.dispatch import DataFrameType\n \n from .operator import ColumnSelector, Operator\n@@ -39,3 +40,38 @@\n     @property\n     def output_properties(self):\n         return self.properties\n+\n+\n+class AddTags(AddMetadata):\n+    def __init__(self, tags=None):\n+        super().__init__(tags=tags)\n+\n+\n+class AddProperties(AddMetadata):\n+    def __init__(self, properties=None):\n+        super().__init__(properties=properties)\n+\n+\n+# Wrappers for common features\n+class TagAsUserID(Operator):\n+    @property\n+    def output_tags(self):\n+        return [Tags.USER_ID]\n+\n+\n+class TagAsItemID(Operator):\n+    @property\n+    def output_tags(self):\n+        return [Tags.ITEM_ID]\n+\n+\n+class TagAsUserFeatures(Operator):\n+    @property\n+    def output_tags(self):\n+        return [Tags.USER]\n+\n+\n+class TagAsItemFeatures(Operator):\n+    @property\n+    def output_tags(self):\n+        return [Tags.ITEM]\n", "issue": "[FEA] Simplify AddMetadata Tag for NVTabular\n**Is your feature request related to a problem? Please describe.**\r\nCurrently, we provide the functionality for the user to tag columns with the operator `AddMetadata`.\r\nThe use case is that users will use the operator mainly for adding tags. \r\n\r\nShould we provide a wrapper called `AddTag` or `TagAs` to simplify the AddMetadata operator?\r\nShould we provide multiple wrappers for common tags - e.g.\r\n\r\n`TagAsUserID()`, `TagAsItemID()`, `TagAsUserFeatures()`, `TagAsItemFeatures()`, etc.\r\n\r\n\n", "before_files": [{"content": "#\n# Copyright (c) 2021, NVIDIA CORPORATION.\n#\n# Licensed under the Apache License, Version 2.0 (the \"License\");\n# you may not use this file except in compliance with the License.\n# You may obtain a copy of the License at\n#\n#     http://www.apache.org/licenses/LICENSE-2.0\n#\n# Unless required by applicable law or agreed to in writing, software\n# distributed under the License is distributed on an \"AS IS\" BASIS,\n# WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.\n# See the License for the specific language governing permissions and\n# limitations under the License.\n#\nfrom nvtabular.dispatch import DataFrameType\n\nfrom .operator import ColumnSelector, Operator\n\n\nclass AddMetadata(Operator):\n    \"\"\"\n    This operator will add user defined tags and properties\n    to a Schema.\n    \"\"\"\n\n    def __init__(self, tags=None, properties=None):\n        super().__init__()\n        self.tags = tags or []\n        self.properties = properties or {}\n\n    def transform(self, col_selector: ColumnSelector, df: DataFrameType) -> DataFrameType:\n        return df\n\n    @property\n    def output_tags(self):\n        return self.tags\n\n    @property\n    def output_properties(self):\n        return self.properties\n", "path": "nvtabular/ops/add_metadata.py"}, {"content": "#\n# Copyright (c) 2021, NVIDIA CORPORATION.\n#\n# Licensed under the Apache License, Version 2.0 (the \"License\");\n# you may not use this file except in compliance with the License.\n# You may obtain a copy of the License at\n#\n#     http://www.apache.org/licenses/LICENSE-2.0\n#\n# Unless required by applicable law or agreed to in writing, software\n# distributed under the License is distributed on an \"AS IS\" BASIS,\n# WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.\n# See the License for the specific language governing permissions and\n# limitations under the License.\n#\n\n# alias submodules here to avoid breaking everything with moving to submodules\n# flake8: noqa\nfrom .add_metadata import AddMetadata\nfrom .bucketize import Bucketize\nfrom .categorify import Categorify, get_embedding_sizes\nfrom .clip import Clip\nfrom .column_similarity import ColumnSimilarity\nfrom .data_stats import DataStats\nfrom .difference_lag import DifferenceLag\nfrom .drop_low_cardinality import DropLowCardinality\nfrom .dropna import Dropna\nfrom .fill import FillMedian, FillMissing\nfrom .filter import Filter\nfrom .groupby import Groupby\nfrom .hash_bucket import HashBucket\nfrom .hashed_cross import HashedCross\nfrom .join_external import JoinExternal\nfrom .join_groupby import JoinGroupby\nfrom .lambdaop import LambdaOp\nfrom .list_slice import ListSlice\nfrom .logop import LogOp\nfrom .normalize import Normalize, NormalizeMinMax\nfrom .operator import ColumnSelector, Operator\nfrom .reduce_dtype_size import ReduceDtypeSize\nfrom .rename import Rename\nfrom .stat_operator import StatOperator\nfrom .target_encoding import TargetEncoding\nfrom .value_counts import ValueCount\n", "path": "nvtabular/ops/__init__.py"}], "after_files": [{"content": "#\n# Copyright (c) 2021, NVIDIA CORPORATION.\n#\n# Licensed under the Apache License, Version 2.0 (the \"License\");\n# you may not use this file except in compliance with the License.\n# You may obtain a copy of the License at\n#\n#     http://www.apache.org/licenses/LICENSE-2.0\n#\n# Unless required by applicable law or agreed to in writing, software\n# distributed under the License is distributed on an \"AS IS\" BASIS,\n# WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.\n# See the License for the specific language governing permissions and\n# limitations under the License.\n#\nfrom merlin.schema.tags import Tags\nfrom nvtabular.dispatch import DataFrameType\n\nfrom .operator import ColumnSelector, Operator\n\n\nclass AddMetadata(Operator):\n    \"\"\"\n    This operator will add user defined tags and properties\n    to a Schema.\n    \"\"\"\n\n    def __init__(self, tags=None, properties=None):\n        super().__init__()\n        self.tags = tags or []\n        self.properties = properties or {}\n\n    def transform(self, col_selector: ColumnSelector, df: DataFrameType) -> DataFrameType:\n        return df\n\n    @property\n    def output_tags(self):\n        return self.tags\n\n    @property\n    def output_properties(self):\n        return self.properties\n\n\nclass AddTags(AddMetadata):\n    def __init__(self, tags=None):\n        super().__init__(tags=tags)\n\n\nclass AddProperties(AddMetadata):\n    def __init__(self, properties=None):\n        super().__init__(properties=properties)\n\n\n# Wrappers for common features\nclass TagAsUserID(Operator):\n    @property\n    def output_tags(self):\n        return [Tags.USER_ID]\n\n\nclass TagAsItemID(Operator):\n    @property\n    def output_tags(self):\n        return [Tags.ITEM_ID]\n\n\nclass TagAsUserFeatures(Operator):\n    @property\n    def output_tags(self):\n        return [Tags.USER]\n\n\nclass TagAsItemFeatures(Operator):\n    @property\n    def output_tags(self):\n        return [Tags.ITEM]\n", "path": "nvtabular/ops/add_metadata.py"}, {"content": "#\n# Copyright (c) 2021, NVIDIA CORPORATION.\n#\n# Licensed under the Apache License, Version 2.0 (the \"License\");\n# you may not use this file except in compliance with the License.\n# You may obtain a copy of the License at\n#\n#     http://www.apache.org/licenses/LICENSE-2.0\n#\n# Unless required by applicable law or agreed to in writing, software\n# distributed under the License is distributed on an \"AS IS\" BASIS,\n# WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.\n# See the License for the specific language governing permissions and\n# limitations under the License.\n#\n\n# alias submodules here to avoid breaking everything with moving to submodules\n# flake8: noqa\nfrom .add_metadata import (\n    AddMetadata,\n    AddProperties,\n    AddTags,\n    TagAsItemFeatures,\n    TagAsItemID,\n    TagAsUserFeatures,\n    TagAsUserID,\n)\nfrom .bucketize import Bucketize\nfrom .categorify import Categorify, get_embedding_sizes\nfrom .clip import Clip\nfrom .column_similarity import ColumnSimilarity\nfrom .data_stats import DataStats\nfrom .difference_lag import DifferenceLag\nfrom .drop_low_cardinality import DropLowCardinality\nfrom .dropna import Dropna\nfrom .fill import FillMedian, FillMissing\nfrom .filter import Filter\nfrom .groupby import Groupby\nfrom .hash_bucket import HashBucket\nfrom .hashed_cross import HashedCross\nfrom .join_external import JoinExternal\nfrom .join_groupby import JoinGroupby\nfrom .lambdaop import LambdaOp\nfrom .list_slice import ListSlice\nfrom .logop import LogOp\nfrom .normalize import Normalize, NormalizeMinMax\nfrom .operator import ColumnSelector, Operator\nfrom .reduce_dtype_size import ReduceDtypeSize\nfrom .rename import Rename\nfrom .stat_operator import StatOperator\nfrom .target_encoding import TargetEncoding\nfrom .value_counts import ValueCount\n", "path": "nvtabular/ops/__init__.py"}]}