Remove _add_docstring_header decorator from amazon review polarity (#…

…1611) Co-authored-by: nayef211 <n63ahmed@edu.uwaterloo.ca>
pytorch · Feb 14, 2022 · 8808e7e · 8808e7e
1 parent 7937703
commit 8808e7e
Showing 1 changed file with 0 additions and 2 deletions.
diff --git a/torchtext/datasets/amazonreviewpolarity.py b/torchtext/datasets/amazonreviewpolarity.py
@@ -4,7 +4,6 @@
 from torchtext._internal.module_utils import is_module_available
 from torchtext.data.datasets_utils import (
     _wrap_split_argument,
-    _add_docstring_header,
     _create_dataset_directory,
 )
 
@@ -32,7 +31,6 @@
 DATASET_NAME = "AmazonReviewPolarity"
 
 
-@_add_docstring_header(num_lines=NUM_LINES, num_classes=2)
 @_create_dataset_directory(dataset_name=DATASET_NAME)
 @_wrap_split_argument(("train", "test"))
 def AmazonReviewPolarity(root: str, split: Union[Tuple[str], str]):