Archives/langchain
2
0
Fork 1
mirror of https://github.com/hwchase17/langchain synced 2024-11-04 06:00:26 +00:00
Code Issues Packages Projects Releases Wiki Activity

fix mypy complaint

Browse Source
This commit is contained in:
olgavrou 2023-08-29 05:51:52 -04:00
parent d50c0f139d
commit 4e6e03ef50
1 changed files with 2 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

4
libs/langchain/tests/unit_tests/chains/rl_chain/test_rl_chain_base_embedder.py
View File

@ -1,4 +1,4 @@
from typing import List
from typing import List, Union
import pytest
from test_utils import MockEncoder
@ -118,7 +118,7 @@ def test_simple_action_strlist_no_emb() -> None:
str2 = "test2"
str3 = "test3"
expected = [{"a_namespace": str1}, {"a_namespace": str2}, {"a_namespace": str3}]
to_embed: List[str] = [str1, str2, str3]
to_embed: List[Union[str, base._Embed]] = [str1, str2, str3]
assert base.embed(to_embed, MockEncoder(), "a_namespace") == expected