Error while iterating over SimilarityABC with chunk.shape[0] == 1

[hg-bauerc]

Having this code section of SimilarityABC.iter

if chunk.shape[0] > 1:
    for sim in self[chunk]:
        yield sim
else:
    yield self[chunk]

When running this code and chunk.shape[0] == 1 (i.e. else path) I get a ValueError (too many values to unpack).
If I'm right the above code has to be simply replaced by

for sim in self[chunk]:
    yield sim

Thanks for verifying!

[hg-bauerc]

The problem with the else-path can be seen in this demo session:

>>> from gensim import corpora, models, similarities
>>> corpus = [[(0, 1.0), (1, 1.0), (2, 1.0)],
...           [(2, 1.0), (3, 1.0), (4, 1.0), (5, 1.0), (6, 1.0), (8, 1.0)],
...           [(1, 1.0), (3, 1.0), (4, 1.0), (7, 1.0)],
...           [(0, 1.0), (4, 2.0), (7, 1.0)]]
>>> tfidf = models.TfidfModel(corpus)
>>> index = similarities.MatrixSimilarity(tfidf[corpus], num_features=12)
>>> index.chunksize=2
>>> for s in index:
...   print s
...
[ 0.99999994  0.15336274  0.32415688  0.35208049]
[ 0.15336274  1.          0.17483118  0.05580689]
[ 0.32415688  0.17483118  1.00000012  0.46034482]
[ 0.35208049  0.05580689  0.46034482  1.        ]
>>> index.chunksize=3
>>> for s in index:
...   print s
...
[ 0.99999994  0.15336274  0.32415688  0.35208049]
[ 0.15336274  1.          0.17483118  0.05580689]
[ 0.32415688  0.17483118  1.00000012  0.46034482]
[[ 0.35208049  0.05580689  0.46034482  1.        ]]

With chunksize == 3 the else-path is called for the last sim-vector and returns it inside a list.