Fix minimaxir#57 by not wrapping in a numpy array

AhLeng · Aug 4, 2018 · 6bce18c · 6bce18c
1 parent 42e79fd
commit 6bce18c
Show file tree

Hide file tree

Showing 2 changed files with 1 addition and 2 deletions.
diff --git 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
@@ -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'])