Fix bug in Similarity.query_shards in multiprocessing case (#2044)

piskvorky · Jun 22, 2018 · fd6f6eb · fd6f6eb
1 parent cc441b7
commit fd6f6eb
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/gensim/similarities/docsim.py b/gensim/similarities/docsim.py
@@ -462,7 +462,7 @@ def query_shards(self, query):
         if PARALLEL_SHARDS and PARALLEL_SHARDS > 1:
             logger.debug("spawning %i query processes", PARALLEL_SHARDS)
             pool = multiprocessing.Pool(PARALLEL_SHARDS)
-            result = pool.imap(query_shard, args, chunksize=1 + len(args) / PARALLEL_SHARDS)
+            result = pool.imap(query_shard, args, chunksize=1 + len(list(args)) / PARALLEL_SHARDS)
         else:
             # serial processing, one shard after another
             pool = None