From 37063469cefee1d721517aa6420b078fa6f20203 Mon Sep 17 00:00:00 2001 From: =?UTF-8?q?Radim=20=C5=98eh=C5=AF=C5=99ek?= Date: Mon, 19 Oct 2020 01:54:27 +0200 Subject: [PATCH] get rid of forgotten itervalues --- gensim/corpora/sharded_corpus.py | 3 +-- 1 file changed, 1 insertion(+), 2 deletions(-) diff --git a/gensim/corpora/sharded_corpus.py b/gensim/corpora/sharded_corpus.py index fa3b76ad0e..92729bb245 100644 --- 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: