Errors? when initial idf values in testing part. #13

Hi zhuyun,
Please help me check this problem:

I think when 'with_idf=True', this line should be:
if len(cols) < 3:
idf = np.ones(len(q))
else:
idf = np.array([int(t) for t in cols[2].split(',')])

all right?

Thanks for you attention.