fix dataset retrival in dataset mode (#3334)

This commit is contained in:
Jyong
2024-04-11 02:11:21 +08:00
committed by GitHub
parent 826c422ac4
commit 6164604462
2 changed files with 93 additions and 0 deletions

View File

@@ -34,6 +34,7 @@ class CSVExtractor(BaseExtractor):
def extract(self) -> list[Document]:
"""Load data into document objects."""
docs = []
try:
with open(self._file_path, newline="", encoding=self._encoding) as csvfile:
docs = self._read_from_file(csvfile)