From e6af66fcb3acf09c95695f436cc7a5bb0320fdd4 Mon Sep 17 00:00:00 2001
From: Tom Aarsen <37621491+tomaarsen@users.noreply.github.com>
Date: Tue, 19 Mar 2024 08:42:58 +0100
Subject: [PATCH] Also update return docstring of encode_multi_process (#2548)

---
 sentence_transformers/SentenceTransformer.py | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/sentence_transformers/SentenceTransformer.py b/sentence_transformers/SentenceTransformer.py
index 3d7869ace..ae45ec002 100644
--- a/sentence_transformers/SentenceTransformer.py
+++ b/sentence_transformers/SentenceTransformer.py
@@ -476,7 +476,7 @@ def encode_multi_process(
         :param chunk_size: Sentences are chunked and sent to the individual processes. If none, it determine a sensible size.
         :param normalize_embeddings: Whether to normalize returned vectors to have length 1. In that case,
             the faster dot-product (util.dot_score) instead of cosine similarity can be used.
-        :return: Numpy matrix with all embeddings
+        :return: 2d numpy array with shape [num_inputs, output_dimension]
         """
 
         if chunk_size is None: