From fb3aa67c31d06caffc250abd374ef03621021f71 Mon Sep 17 00:00:00 2001 From: Ritesh Agarwal Date: Tue, 28 Jan 2025 19:49:06 -0800 Subject: [PATCH] Randomly pick similarity when not provided explicitly. Change-Id: I8ca21ed5c11ec0f8dd6bc1e20522d059c52cb25b Reviewed-on: https://review.couchbase.org/c/TAF/+/222557 Tested-by: Build Bot Reviewed-by: --- pytests/aGoodDoctor/n1ql.py | 4 +++- 1 file changed, 3 insertions(+), 1 deletion(-) diff --git a/pytests/aGoodDoctor/n1ql.py b/pytests/aGoodDoctor/n1ql.py index 8619fe0af..382fb05be 100644 --- a/pytests/aGoodDoctor/n1ql.py +++ b/pytests/aGoodDoctor/n1ql.py @@ -324,7 +324,9 @@ def create_indexes(self, buckets, skip_index=False, base64=False, xattr=False): if workload.get("vector"): vector_defn = workload.get("vector")[vector_defn_counter % len(workload.get("vector"))] dim = workload.get("dim") - similarity = vector_defn["similarity"] + similarity = TestInputSingleton.input.param( + "similarity", + random.choice("L2_SQUARED", "EUCLIDEAN_SQUARED", "COSINE", "DOT")) nProbe = vector_defn["nProbe"] description = vector_defn.get("description", "IVF,%s" % vector_defn.get("quantization", "SQ8")) vector_fields = "'dimension': {}, 'description': '{}', 'similarity': '{}', 'scan_nprobes': {}".format(