github/langchain
1
0
Fork 0
mirror of https://github.com/hwchase17/langchain.git synced 2025-07-07 13:40:46 +00:00
Code Issues Packages Projects Releases Wiki Activity

make input arg type more explicit

Browse Source
This commit is contained in:
olgavrou 2023-08-29 05:42:45 -04:00
parent 8d10a52525
commit 44485c2b26
1 changed files with 2 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

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

@ -116,7 +116,8 @@ 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}]
assert base.embed([str1, str2, str3], MockEncoder(), "a_namespace") == expected to_embed: List[str] = [str1, str2, str3]
assert base.embed(to_embed, MockEncoder(), "a_namespace") == expected
@pytest.mark.requires("vowpal_wabbit_next") @pytest.mark.requires("vowpal_wabbit_next")