Skip to content

Commit 6894f48

Browse files
Update tests/system/test_openai_encoder.py
Co-authored-by: igiloh-pinecone <118673156+igiloh-pinecone@users.noreply.github.com>
1 parent e8ae3fc commit 6894f48

File tree

1 file changed

+2
-5
lines changed

1 file changed

+2
-5
lines changed

tests/system/test_openai_encoder.py

Lines changed: 2 additions & 5 deletions
Original file line numberDiff line numberDiff line change
@@ -24,11 +24,8 @@ def test_init_with_kwargs():
2424

2525

2626
def encode_by_type(openai_encoder, encoding_function, test_input):
27-
if encoding_function == "encode_documents":
28-
return openai_encoder.encode_documents(test_input)
29-
elif encoding_function == "encode_queries":
30-
return openai_encoder.encode_queries(test_input)
31-
pytest.fail(f"Unknown encoding function: {encoding_function}")
27+
func = getattr(openai_encoder, encoding_function)
28+
return func(test_input)
3229

3330

3431
@pytest.mark.parametrize(

0 commit comments

Comments
 (0)