From 4db4839995be0bad5a75da363cb8738b0a4d6412 Mon Sep 17 00:00:00 2001 From: Nirant Date: Wed, 20 Mar 2024 19:11:44 +0530 Subject: [PATCH] [PyPi Publish] Bump version to 0.2.5 in pyproject.toml (#156) * Bump version to 0.2.5 in pyproject.toml * chore: case insensitive check (#157) --- fastembed/sparse/sparse_text_embedding.py | 2 +- fastembed/text/text_embedding.py | 2 +- pyproject.toml | 2 +- 3 files changed, 3 insertions(+), 3 deletions(-) diff --git a/fastembed/sparse/sparse_text_embedding.py b/fastembed/sparse/sparse_text_embedding.py index e5b6b265..aa838ee8 100644 --- a/fastembed/sparse/sparse_text_embedding.py +++ b/fastembed/sparse/sparse_text_embedding.py @@ -48,7 +48,7 @@ def __init__( for EMBEDDING_MODEL_TYPE in self.EMBEDDINGS_REGISTRY: supported_models = EMBEDDING_MODEL_TYPE.list_supported_models() - if any(model_name == model["model"] for model in supported_models): + if any(model_name.lower() == model["model"].lower() for model in supported_models): self.model = EMBEDDING_MODEL_TYPE(model_name, cache_dir, threads, **kwargs) return diff --git a/fastembed/text/text_embedding.py b/fastembed/text/text_embedding.py index cadc9aca..10b43625 100644 --- a/fastembed/text/text_embedding.py +++ b/fastembed/text/text_embedding.py @@ -55,7 +55,7 @@ def __init__( for EMBEDDING_MODEL_TYPE in self.EMBEDDINGS_REGISTRY: supported_models = EMBEDDING_MODEL_TYPE.list_supported_models() - if any(model_name == model["model"] for model in supported_models): + if any(model_name.lower() == model["model"].lower() for model in supported_models): self.model = EMBEDDING_MODEL_TYPE(model_name, cache_dir, threads, **kwargs) return diff --git a/pyproject.toml b/pyproject.toml index 261ca776..446ac594 100644 --- a/pyproject.toml +++ b/pyproject.toml @@ -1,6 +1,6 @@ [tool.poetry] name = "fastembed" -version = "0.2.4" +version = "0.2.5" description = "Fast, light, accurate library built for retrieval embedding generation" authors = ["NirantK "] license = "Apache License"