Mirrors/the-algorithm
2
0
Fork 0
mirror of https://github.com/twitter/the-algorithm.git synced 2024-06-14 15:18:55 +02:00
Code Issues Packages Projects Releases Wiki Activity

Change branching by if statement to branching by match statement

Browse Source
This commit is contained in:
Yuki 2023-04-07 19:28:51 +09:00
parent ec83d01dca
commit 8fc5ef9a62
1 changed files with 11 additions and 10 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

21
ann/src/main/python/dataflow/faiss_index_bq_dataset.py
View File

@ -82,16 +82,17 @@ def get_bq_query():
def parse_metric(config):
metric_str = config["metric"].lower()
if metric_str == "l2":
return faiss.METRIC_L2
elif metric_str == "ip":
return faiss.METRIC_INNER_PRODUCT
elif metric_str == "l1":
return faiss.METRIC_L1
elif metric_str == "linf":
return faiss.METRIC_Linf
else:
raise Exception(f"Uknown metric: {metric_str}")
match metric_str:
case "12":
return faiss.METRIC_L2
case "ip":
return faiss.METRIC_INNER_PRODUCT
case "l1":
return faiss.METRIC_L1
case "linf":
return faiss.METRIC_Linf
case _:
raise Exception(f"Uknown metric: {metric_str}")
def run_pipeline(argv=[]):