Kevin Hu commited on
Commit
6c993fc
·
1 Parent(s): 2acb50d

Fix csv reader exception. (#4628)

Browse files

### What problem does this PR solve?

#4552
### Type of change

- [x] Bug Fix (non-breaking change which fixes an issue)

Files changed (1) hide show
  1. rag/app/qa.py +1 -1
rag/app/qa.py CHANGED
@@ -394,7 +394,7 @@ def chunk(filename, binary=None, lang="Chinese", callback=None, **kwargs):
394
  f"{len(fails)} failure, line: %s..." % (",".join(fails[:3])) if fails else "")))
395
 
396
  if question:
397
- res.append(beAdoc(deepcopy(doc), question, answer, eng, len(reader)))
398
 
399
  callback(0.6, ("Extract Q&A: {}".format(len(res)) + (
400
  f"{len(fails)} failure, line: %s..." % (",".join(fails[:3])) if fails else "")))
 
394
  f"{len(fails)} failure, line: %s..." % (",".join(fails[:3])) if fails else "")))
395
 
396
  if question:
397
+ res.append(beAdoc(deepcopy(doc), question, answer, eng, len(list(reader))))
398
 
399
  callback(0.6, ("Extract Q&A: {}".format(len(res)) + (
400
  f"{len(fails)} failure, line: %s..." % (",".join(fails[:3])) if fails else "")))