Merge pull request ExpediaGroup#32 from ExpediaGroup/tipike/bugfix

use replace instead of strip
wparsley-eg · Sep 6, 2023 · 6e17017 · 6e17017
2 parents 24ffa98 + dede96b
commit 6e17017
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/sdk/python/feast/expediagroup/vectordb/milvus_online_store.py b/sdk/python/feast/expediagroup/vectordb/milvus_online_store.py
@@ -384,7 +384,7 @@ def _create_index_params(self, tags: Dict[str, str], data_type: DataType):
         (Dict): a dictionary formatted for the create_index params argument
         """
         valid_indexes = IndexType._member_map_
-        index_type_tag = tags.get("index_type", "").upper().strip("BIN_")
+        index_type_tag = tags.get("index_type", "").upper().replace("BIN_", "")
 
         index_type = (
             IndexType[index_type_tag]