Skip to content

Commit f55a410

Browse files
committed
Fix error message
1 parent 69a808c commit f55a410

File tree

2 files changed

+4
-8
lines changed

2 files changed

+4
-8
lines changed

django_mongodb_backend/indexes.py

+3-4
Original file line numberDiff line numberDiff line change
@@ -232,19 +232,18 @@ def check(self, model, connection):
232232
)
233233
)
234234
if isinstance(self.similarities, list) and expected_similarities != len(self.similarities):
235-
given_similarities = len(self.similarities)
236235
similarity_function_text = (
237-
"similarities functions" if given_similarities != 1 else "similarity function"
236+
"similarities functions" if expected_similarities != 1 else "similarity function"
238237
)
239238
errors.append(
240239
Error(
241240
f"An Atlas vector search index requires the same number of similarities and "
242241
f"vector fields, but {expected_similarities} "
243242
f"{similarity_function_text} were expected and "
244-
f"{given_similarities} {'were' if given_similarities != 1 else 'was'} "
243+
f"{len(self.similarities)} {'were' if len(self.similarities) != 1 else 'was'} "
245244
"provided.",
246245
obj=self,
247-
id=f"{self._error_id_prefix}.E006",
246+
id=f"{self._error_id_prefix}.E005",
248247
)
249248
)
250249
return errors

tests/indexes_/test_atlas_indexes.py

+1-4
Original file line numberDiff line numberDiff line change
@@ -114,10 +114,7 @@ def test_invalid_similarity_function(self):
114114

115115
@skipUnlessDBFeature("supports_atlas_search")
116116
def test_deconstruct_default_similarity(self):
117-
index = VectorSearchIndex(
118-
name="recent_article_idx",
119-
fields=["number"],
120-
)
117+
index = VectorSearchIndex(name="recent_article_idx", fields=["number"])
121118
name, args, kwargs = index.deconstruct()
122119
new = VectorSearchIndex(*args, **kwargs)
123120
self.assertEqual(new.similarities, index.similarities)

0 commit comments

Comments
 (0)