From 1670b76bacb2ea1658d1b82f1470f763df49dc8a Mon Sep 17 00:00:00 2001 From: Filipe de Avila Belbute Peres Date: Thu, 19 Jan 2023 10:27:24 -0800 Subject: [PATCH] Convert back to list only after all np operations --- examples/Embedding_long_inputs.ipynb | 3 ++- 1 file changed, 2 insertions(+), 1 deletion(-) diff --git a/examples/Embedding_long_inputs.ipynb b/examples/Embedding_long_inputs.ipynb index fe6b5fab..e4460f47 100644 --- a/examples/Embedding_long_inputs.ipynb +++ b/examples/Embedding_long_inputs.ipynb @@ -204,8 +204,9 @@ " chunk_embeddings.append(get_embedding(chunk, model=model))\n", "\n", " if average:\n", - " chunk_embeddings = np.average(chunk_embeddings, axis=0, weights=[len(c) for c in chunk_embeddings]).tolist()\n", + " chunk_embeddings = np.average(chunk_embeddings, axis=0, weights=[len(c) for c in chunk_embeddings])\n", " chunk_embeddings = chunk_embeddings / np.linalg.norm(chunk_embeddings) # normalizes length to 1\n", + " chunk_embeddings = chunk_embeddings.tolist()\n", " return chunk_embeddings" ] },