Skip to content

Commit f9d8020

Browse files
committed
Fix decode fn in shakespeare_char/prepare.py
1 parent 3611338 commit f9d8020

1 file changed

Lines changed: 1 addition & 1 deletion

File tree

data/shakespeare_char/prepare.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -31,7 +31,7 @@
3131
def encode(s):
3232
return [stoi[c] for c in s] # encoder: take a string, output a list of integers
3333
def decode(l):
34-
''.join([itos[i] for i in l]) # decoder: take a list of integers, output a string
34+
return ''.join([itos[i] for i in l]) # decoder: take a list of integers, output a string
3535

3636
# create the train and test splits
3737
n = len(data)

0 commit comments

Comments
 (0)