aboutsummaryrefslogtreecommitdiff
path: root/dataset.py
diff options
context:
space:
mode:
authorAnthony Wang2022-02-21 15:03:28 -0600
committerAnthony Wang2022-02-21 15:03:28 -0600
commit423c1d8304238888758abdf462d625e88cf7db22 (patch)
treede722ed1eb9ed5d0842577f0d1f7d5a4958ec870 /dataset.py
parent9e84768780946d20ea33deddad6293525b0ba0f9 (diff)
Cleanup and reformat with autopep8
Diffstat (limited to 'dataset.py')
-rw-r--r--dataset.py2
1 files changed, 1 insertions, 1 deletions
diff --git a/dataset.py b/dataset.py
index 53f94cc..193b1fb 100644
--- a/dataset.py
+++ b/dataset.py
@@ -25,4 +25,4 @@ class Dataset(torch.utils.data.Dataset):
def __getitem__(self, index):
return (torch.tensor(self.words_indexes[index:index+self.seq_size]),
- torch.tensor(self.words_indexes[index+1:index+self.seq_size+1])) \ No newline at end of file
+ torch.tensor(self.words_indexes[index+1:index+self.seq_size+1]))