From f9d8020f48b3a670e507bfe6bb0bd26507e71276 Mon Sep 17 00:00:00 2001 From: venusatuluri Date: Sat, 21 Jan 2023 06:14:16 +0000 Subject: [PATCH] Fix decode fn in shakespeare_char/prepare.py --- data/shakespeare_char/prepare.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/data/shakespeare_char/prepare.py b/data/shakespeare_char/prepare.py index 918d1ee..fd95329 100644 --- a/data/shakespeare_char/prepare.py +++ b/data/shakespeare_char/prepare.py @@ -31,7 +31,7 @@ itos = { i:ch for i,ch in enumerate(chars) } def encode(s): return [stoi[c] for c in s] # encoder: take a string, output a list of integers def decode(l): - ''.join([itos[i] for i in l]) # decoder: take a list of integers, output a string + return ''.join([itos[i] for i in l]) # decoder: take a list of integers, output a string # create the train and test splits n = len(data)