Skip to content

Commit

Permalink
Add load() to snowflake loader (#5956)
Browse files Browse the repository at this point in the history
Quick fix for recently added [snowflake data
loader](https://github.com/hwchase17/langchain/pull/5825/files).
  • Loading branch information
rlancemartin authored Jun 9, 2023
1 parent 3678cba commit f3e7ac0
Showing 1 changed file with 3 additions and 0 deletions.
3 changes: 3 additions & 0 deletions langchain/document_loaders/snowflake_loader.py
Original file line number Diff line number Diff line change
Expand Up @@ -118,3 +118,6 @@ def lazy_load(self) -> Iterator[Document]:
metadata = {k: v for k, v in row.items() if k in metadata_columns}
doc = Document(page_content=page_content, metadata=metadata)
yield doc

def load(self) -> List[Document]:
return list(self.lazy_load())

0 comments on commit f3e7ac0

Please sign in to comment.