From 4d6d27cffb224b53537189baa9961526689d6f8e Mon Sep 17 00:00:00 2001 From: NirantK Date: Mon, 25 Sep 2023 16:50:52 +0530 Subject: [PATCH] * test(test_onnx_embeddings.py): remove unnecessary list conversion in embeddings assignment --- tests/test_onnx_embeddings.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/tests/test_onnx_embeddings.py b/tests/test_onnx_embeddings.py index 9c51b713..fa373b31 100644 --- a/tests/test_onnx_embeddings.py +++ b/tests/test_onnx_embeddings.py @@ -7,7 +7,7 @@ def test_default_embedding(): model = DefaultEmbedding() docs = ["hello world", "flag embedding"] - embeddings = np.array(list(model.embed(docs))) + embeddings = np.array(model.embed(docs)) assert embeddings.shape == (2, 384) test_default_embedding() \ No newline at end of file