Add type checking for np arrays

:
achyudh · Dec 19, 2018 · fde81b3 · fde81b3
1 parent 3cccfaf
commit fde81b3
Showing 1 changed file with 1 addition and 0 deletions.
diff --git a/lib/data/util.py b/lib/data/util.py
@@ -99,6 +99,7 @@ def reverse_index(indexed_data, vocab, ravel=False):
     reversed_data = list()
     indexed_data = np.argmax(indexed_data, axis=-1)
     word_idx = {id: word for word, id in vocab.items()}
+    if isinstance(indexed_data, np.ndarray): indexed_data = indexed_data.tolist()
     for indexed_line in indexed_data:
         if ravel:
             reversed_data.append([' '.join((word_idx[x] for x in indexed_line[1:len(indexed_line)-1]))])