From 6bce18c218b08d86a3562318df2d1af35629404b Mon Sep 17 00:00:00 2001 From: Max Woolf Date: Fri, 3 Aug 2018 21:27:33 -0700 Subject: [PATCH] Fix #57 by not wrapping in a numpy array --- setup.py | 2 +- textgenrnn/model_training.py | 1 - 2 files changed, 1 insertion(+), 2 deletions(-) diff --git a/setup.py b/setup.py index 0844a36..c904a5f 100644 --- a/setup.py +++ b/setup.py @@ -25,7 +25,7 @@ setup( name='textgenrnn', packages=['textgenrnn'], # this must be the same as the name above - version='1.3.1', + version='1.3.2', description='Easily train your own text-generating neural network ' \ 'of any size and complexity', long_description=long_description, diff --git a/textgenrnn/model_training.py b/textgenrnn/model_training.py index 34683ae..91727d8 100644 --- a/textgenrnn/model_training.py +++ b/textgenrnn/model_training.py @@ -75,7 +75,6 @@ def generate_sequences_from_texts(texts, indices_list, def process_sequence(X, textgenrnn, new_tokenizer): - X = np.array(X) X = new_tokenizer.texts_to_sequences(X) X = sequence.pad_sequences( X, maxlen=textgenrnn.config['max_length'])