From 47447b6f9846fb63cc17d3f458df405387f46127 Mon Sep 17 00:00:00 2001 From: Patrick Carlson Date: Sat, 23 Dec 2023 15:46:19 -0600 Subject: [PATCH] identity to equality check (#1566) ## Description of changes Changes identity `is` to equality `==` check Co-authored-by: Jeffrey Huber --- 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: