From a0a41db7132a350c4f689390beae5b525199be43 Mon Sep 17 00:00:00 2001 From: Patrick Carlson Date: Thu, 21 Dec 2023 22:48:10 -0600 Subject: [PATCH] identity to equality check --- chromadb/utils/embedding_functions.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/chromadb/utils/embedding_functions.py b/chromadb/utils/embedding_functions.py index d3f3f1d2071..b7aeb08bf6b 100644 --- a/chromadb/utils/embedding_functions.py +++ b/chromadb/utils/embedding_functions.py @@ -610,7 +610,7 @@ def __init__( self._model_name = model_name self._task_type = task_type self._task_title = None - if self._task_type is "RETRIEVAL_DOCUMENT": + if self._task_type == "RETRIEVAL_DOCUMENT": self._task_title = "Embedding of single string" def __call__(self, input: Documents) -> Embeddings: