fixing stylecheck (#1247)

parmeet · web-flow · commit d98aee07dd73 · 2021-03-08T11:36:00.000-05:00
diff --git a/torchtext/datasets/ag_news.py b/torchtext/datasets/ag_news.py
@@ -26,6 +26,8 @@
 }
 
 DATASET_NAME = "AG_NEWS"
+
+
 @_add_docstring_header(num_lines=NUM_LINES, num_classes=4)
 @_create_dataset_directory(dataset_name=DATASET_NAME)
 @_wrap_split_argument(('train', 'test'))
diff --git a/torchtext/datasets/amazonreviewfull.py b/torchtext/datasets/amazonreviewfull.py
@@ -31,6 +31,8 @@
 }
 
 DATASET_NAME = "AmazonReviewFull"
+
+
 @_add_docstring_header(num_lines=NUM_LINES, num_classes=5)
 @_create_dataset_directory(dataset_name=DATASET_NAME)
 @_wrap_split_argument(('train', 'test'))
diff --git a/torchtext/datasets/amazonreviewpolarity.py b/torchtext/datasets/amazonreviewpolarity.py
@@ -31,6 +31,8 @@
 }
 
 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'))
diff --git a/torchtext/datasets/conll2000chunking.py b/torchtext/datasets/conll2000chunking.py
@@ -36,6 +36,8 @@
 
 
 DATASET_NAME = "CoNLL2000Chunking"
+
+
 @_add_docstring_header(num_lines=NUM_LINES)
 @_create_dataset_directory(dataset_name=DATASET_NAME)
 @_wrap_split_argument(('train', 'test'))
diff --git a/torchtext/datasets/dbpedia.py b/torchtext/datasets/dbpedia.py
@@ -24,6 +24,8 @@
 _PATH = 'dbpedia_csv.tar.gz'
 
 DATASET_NAME = "DBpedia"
+
+
 @_add_docstring_header(num_lines=NUM_LINES, num_classes=14)
 @_create_dataset_directory(dataset_name=DATASET_NAME)
 @_wrap_split_argument(('train', 'test'))
diff --git a/torchtext/datasets/enwik9.py b/torchtext/datasets/enwik9.py
@@ -20,6 +20,8 @@
 }
 
 DATASET_NAME = "EnWik9"
+
+
 @_add_docstring_header(num_lines=NUM_LINES)
 @_create_dataset_directory(dataset_name=DATASET_NAME)
 @_wrap_split_argument(('train',))
diff --git a/torchtext/datasets/imdb.py b/torchtext/datasets/imdb.py
@@ -17,6 +17,8 @@
 _PATH = 'aclImdb_v1.tar.gz'
 
 DATASET_NAME = "IMDB"
+
+
 @_add_docstring_header(num_lines=NUM_LINES, num_classes=2)
 @_create_dataset_directory(dataset_name=DATASET_NAME)
 @_wrap_split_argument(('train', 'test'))
diff --git a/torchtext/datasets/iwslt2016.py b/torchtext/datasets/iwslt2016.py
@@ -143,6 +143,8 @@ def _construct_filepaths(paths, src_filename, tgt_filename):
 
 
 DATASET_NAME = "IWSLT2016"
+
+
 @_create_dataset_directory(dataset_name=DATASET_NAME)
 @_wrap_split_argument(('train', 'valid', 'test'))
 def IWSLT2016(root='.data', split=('train', 'valid', 'test'), language_pair=('de', 'en'), valid_set='tst2013', test_set='tst2014'):
diff --git a/torchtext/datasets/iwslt2017.py b/torchtext/datasets/iwslt2017.py
@@ -115,6 +115,8 @@ def _construct_filepaths(paths, src_filename, tgt_filename):
 
 
 DATASET_NAME = "IWSLT2017"
+
+
 @_create_dataset_directory(dataset_name=DATASET_NAME)
 @_wrap_split_argument(('train', 'valid', 'test'))
 def IWSLT2017(root='.data', split=('train', 'valid', 'test'), language_pair=('de', 'en')):
diff --git a/torchtext/datasets/penntreebank.py b/torchtext/datasets/penntreebank.py
@@ -27,6 +27,8 @@
 }
 
 DATASET_NAME = "PennTreebank"
+
+
 @_add_docstring_header(num_lines=NUM_LINES)
 @_create_dataset_directory(dataset_name=DATASET_NAME)
 @_wrap_split_argument(('train', 'valid', 'test'))
diff --git a/torchtext/datasets/sogounews.py b/torchtext/datasets/sogounews.py
@@ -31,6 +31,8 @@
 }
 
 DATASET_NAME = "SogouNews"
+
+
 @_add_docstring_header(num_lines=NUM_LINES, num_classes=5)
 @_create_dataset_directory(dataset_name=DATASET_NAME)
 @_wrap_split_argument(('train', 'test'))
diff --git a/torchtext/datasets/squad1.py b/torchtext/datasets/squad1.py
@@ -23,6 +23,8 @@
 
 
 DATASET_NAME = "SQuAD1"
+
+
 @_add_docstring_header(num_lines=NUM_LINES)
 @_create_dataset_directory(dataset_name=DATASET_NAME)
 @_wrap_split_argument(('train', 'dev'))
diff --git a/torchtext/datasets/squad2.py b/torchtext/datasets/squad2.py
@@ -23,6 +23,8 @@
 
 
 DATASET_NAME = "SQuAD2"
+
+
 @_add_docstring_header(num_lines=NUM_LINES)
 @_create_dataset_directory(dataset_name=DATASET_NAME)
 @_wrap_split_argument(('train', 'dev'))
diff --git a/torchtext/datasets/udpos.py b/torchtext/datasets/udpos.py
@@ -20,6 +20,8 @@
 
 
 DATASET_NAME = "UDPOS"
+
+
 @_add_docstring_header(num_lines=NUM_LINES)
 @_create_dataset_directory(dataset_name=DATASET_NAME)
 @_wrap_split_argument(('train', 'valid', 'test'))
diff --git a/torchtext/datasets/wikitext103.py b/torchtext/datasets/wikitext103.py
@@ -23,6 +23,8 @@
 }
 
 DATASET_NAME = "WikiText103"
+
+
 @_add_docstring_header(num_lines=NUM_LINES)
 @_create_dataset_directory(dataset_name=DATASET_NAME)
 @_wrap_split_argument(('train', 'valid', 'test'))
diff --git a/torchtext/datasets/wikitext2.py b/torchtext/datasets/wikitext2.py
@@ -20,6 +20,8 @@
 }
 
 DATASET_NAME = "WikiText2"
+
+
 @_add_docstring_header(num_lines=NUM_LINES)
 @_create_dataset_directory(dataset_name=DATASET_NAME)
 @_wrap_split_argument(('train', 'valid', 'test'))
diff --git a/torchtext/datasets/yahooanswers.py b/torchtext/datasets/yahooanswers.py
@@ -1,12 +1,11 @@
-from torchtext.utils import download_from_url, extract_archive, unicode_csv_reader
+from torchtext.utils import download_from_url, extract_archive
 from torchtext.data.datasets_utils import _RawTextIterableDataset
 from torchtext.data.datasets_utils import _wrap_split_argument
 from torchtext.data.datasets_utils import _add_docstring_header
 from torchtext.data.datasets_utils import _find_match
 from torchtext.data.datasets_utils import _create_dataset_directory
 from torchtext.data.datasets_utils import _create_data_from_csv
 import os
-import io
 
 URL = 'https://drive.google.com/uc?export=download&id=0Bz8a_Dbh9Qhbd2JNdDBsQUdocVU'
 
@@ -20,6 +19,8 @@
 _PATH = 'yahoo_answers_csv.tar.gz'
 
 DATASET_NAME = "YahooAnswers"
+
+
 @_add_docstring_header(num_lines=NUM_LINES, num_classes=10)
 @_create_dataset_directory(dataset_name=DATASET_NAME)
 @_wrap_split_argument(('train', 'test'))
diff --git a/torchtext/datasets/yelpreviewfull.py b/torchtext/datasets/yelpreviewfull.py
@@ -24,6 +24,8 @@
 _PATH = 'yelp_review_full_csv.tar.gz'
 
 DATASET_NAME = "YelpReviewFull"
+
+
 @_add_docstring_header(num_lines=NUM_LINES, num_classes=5)
 @_create_dataset_directory(dataset_name=DATASET_NAME)
 @_wrap_split_argument(('train', 'test'))
diff --git a/torchtext/datasets/yelpreviewpolarity.py b/torchtext/datasets/yelpreviewpolarity.py
@@ -21,6 +21,8 @@
 _PATH = 'yelp_review_polarity_csv.tar.gz'
 
 DATASET_NAME = "YelpReviewPolarity"
+
+
 @_add_docstring_header(num_lines=NUM_LINES, num_classes=2)
 @_create_dataset_directory(dataset_name=DATASET_NAME)
 @_wrap_split_argument(('train', 'test'))
diff --git a/torchtext/experimental/datasets/raw/multi30k.py b/torchtext/experimental/datasets/raw/multi30k.py
@@ -398,6 +398,8 @@ def _construct_filepaths(paths, src_filename, tgt_filename):
 
 
 DATASET_NAME = "Multi30k"
+
+
 @_create_dataset_directory(dataset_name=DATASET_NAME)
 @_wrap_split_argument(('train', 'valid', 'test'))
 def Multi30k(root, split,
diff --git a/torchtext/experimental/datasets/raw/wmt14.py b/torchtext/experimental/datasets/raw/wmt14.py
@@ -59,6 +59,8 @@ def _construct_filepaths(paths, src_filename, tgt_filename):
 
 
 DATASET_NAME = "WMT14"
+
+
 @_create_dataset_directory(dataset_name=DATASET_NAME)
 @_wrap_split_argument(('train', 'valid', 'test'))
 def WMT14(root, split,
diff --git a/torchtext/experimental/datasets/raw/wmtnewscrawl.py b/torchtext/experimental/datasets/raw/wmtnewscrawl.py
@@ -44,6 +44,8 @@
 }
 
 DATASET_NAME = "WMTNewsCrawl"
+
+
 @_add_docstring_header(num_lines=NUM_LINES)
 @_create_dataset_directory(dataset_name=DATASET_NAME)
 @_wrap_split_argument('train')

Original file line number	Diff line number	Diff line change
`@@ -26,6 +26,8 @@`
`26`	`26`	`}`
`27`	`27`
`28`	`28`	`DATASET_NAME = "AG_NEWS"`
	`29`	`+`
	`30`	`+`
`29`	`31`	`@_add_docstring_header(num_lines=NUM_LINES, num_classes=4)`
`30`	`32`	`@_create_dataset_directory(dataset_name=DATASET_NAME)`
`31`	`33`	`@_wrap_split_argument(('train', 'test'))`
Original file line number	Diff line number	Diff line change
`@@ -31,6 +31,8 @@`
`31`	`31`	`}`
`32`	`32`
`33`	`33`	`DATASET_NAME = "AmazonReviewFull"`
	`34`	`+`
	`35`	`+`
`34`	`36`	`@_add_docstring_header(num_lines=NUM_LINES, num_classes=5)`
`35`	`37`	`@_create_dataset_directory(dataset_name=DATASET_NAME)`
`36`	`38`	`@_wrap_split_argument(('train', 'test'))`
Original file line number	Diff line number	Diff line change
`@@ -20,6 +20,8 @@`
`20`	`20`	`}`
`21`	`21`
`22`	`22`	`DATASET_NAME = "EnWik9"`
	`23`	`+`
	`24`	`+`
`23`	`25`	`@_add_docstring_header(num_lines=NUM_LINES)`
`24`	`26`	`@_create_dataset_directory(dataset_name=DATASET_NAME)`
`25`	`27`	`@_wrap_split_argument(('train',))`
Original file line number	Diff line number	Diff line change
`@@ -27,6 +27,8 @@`
`27`	`27`	`}`
`28`	`28`
`29`	`29`	`DATASET_NAME = "PennTreebank"`
	`30`	`+`
	`31`	`+`
`30`	`32`	`@_add_docstring_header(num_lines=NUM_LINES)`
`31`	`33`	`@_create_dataset_directory(dataset_name=DATASET_NAME)`
`32`	`34`	`@_wrap_split_argument(('train', 'valid', 'test'))`
Original file line number	Diff line number	Diff line change
`@@ -23,6 +23,8 @@`
`23`	`23`	`}`
`24`	`24`
`25`	`25`	`DATASET_NAME = "WikiText103"`
	`26`	`+`
	`27`	`+`
`26`	`28`	`@_add_docstring_header(num_lines=NUM_LINES)`
`27`	`29`	`@_create_dataset_directory(dataset_name=DATASET_NAME)`
`28`	`30`	`@_wrap_split_argument(('train', 'valid', 'test'))`
Original file line number	Diff line number	Diff line change
`@@ -44,6 +44,8 @@`
`44`	`44`	`}`
`45`	`45`
`46`	`46`	`DATASET_NAME = "WMTNewsCrawl"`
	`47`	`+`
	`48`	`+`
`47`	`49`	`@_add_docstring_header(num_lines=NUM_LINES)`
`48`	`50`	`@_create_dataset_directory(dataset_name=DATASET_NAME)`
`49`	`51`	`@_wrap_split_argument('train')`