From f3e7ac0a2c0ad677e91571f59b03b55c5af52db2 Mon Sep 17 00:00:00 2001 From: Lance Martin <122662504+rlancemartin@users.noreply.github.com> Date: Fri, 9 Jun 2023 11:27:29 -0700 Subject: [PATCH] Add load() to snowflake loader (#5956) Quick fix for recently added [snowflake data loader](https://github.com/hwchase17/langchain/pull/5825/files). --- langchain/document_loaders/snowflake_loader.py | 3 +++ 1 file changed, 3 insertions(+) diff --git a/langchain/document_loaders/snowflake_loader.py b/langchain/document_loaders/snowflake_loader.py index 115f6e7b9e..b76a742609 100644 --- a/langchain/document_loaders/snowflake_loader.py +++ b/langchain/document_loaders/snowflake_loader.py @@ -118,3 +118,6 @@ class SnowflakeLoader(BaseLoader): 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())