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: