get rid of forgotten itervalues

piskvorky · Oct 18, 2020 · 3706346 · 3706346
1 parent c231478
commit 3706346
Showing 1 changed file with 1 addition and 2 deletions.
diff --git a/gensim/corpora/sharded_corpus.py b/gensim/corpora/sharded_corpus.py
@@ -288,8 +288,7 @@ def init_shards(self, output_prefix, corpus, shardsize=4096, dtype=_default_dtyp
             logger.debug('Current chunk dimension: %d x %d', len(doc_chunk), self.dim)
 
             for i, doc in enumerate(doc_chunk):
-                doc = dict(doc)
-                current_shard[i][list(doc)] = list(gensim.matutils.itervalues(doc))
+                current_shard[i][list(doc)] = list(dict(doc).values())
 
             # Handles the updating as well.
             if self.sparse_serialization: