From 4db892644ffe48cec216b4b27a09a5078329077d Mon Sep 17 00:00:00 2001 From: nayef211 Date: Mon, 14 Feb 2022 10:52:41 -0800 Subject: [PATCH] Remove _add_docstring_header decorator from amazon review polarity --- torchtext/datasets/amazonreviewpolarity.py | 2 -- 1 file changed, 2 deletions(-) diff --git a/torchtext/datasets/amazonreviewpolarity.py b/torchtext/datasets/amazonreviewpolarity.py index a3c273038e..6055f3a458 100644 --- 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]):