From b9d75d2afaf831da00f890f2b5318887769028d0 Mon Sep 17 00:00:00 2001 From: Eugene Yurtsev Date: Tue, 6 Feb 2024 17:09:18 -0500 Subject: [PATCH] x --- tests/extraction/test_extraction_with_chain.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/tests/extraction/test_extraction_with_chain.py b/tests/extraction/test_extraction_with_chain.py index 63beedd..055e7de 100644 --- a/tests/extraction/test_extraction_with_chain.py +++ b/tests/extraction/test_extraction_with_chain.py @@ -62,7 +62,7 @@ def test_create_extraction_chain_with_csv_encoder(options: Mapping[str, Any]) -> chain = create_extraction_chain(chat_model, **options) assert isinstance(chain, LLMChain) # Try to run through predict and parse - chain.invoke("some string") + chain.invoke("some string") # type: ignore MANY_TEXT_SCHEMA = Text(