github/langchain
1
0
Fork 0
mirror of https://github.com/hwchase17/langchain.git synced 2025-06-02 21:23:32 +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 import pytest
from test_utils import MockEncoder from test_utils import MockEncoder
@ -118,7 +118,7 @@ def test_simple_action_strlist_no_emb() -> None:
str2 = "test2" str2 = "test2"
str3 = "test3" str3 = "test3"
expected = [{"a_namespace": str1}, {"a_namespace": str2}, {"a_namespace": str3}] 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 assert base.embed(to_embed, MockEncoder(), "a_namespace") == expected