check key when sorting Neighbors if distance is equal

aerospike · Oct 11, 2024 · 2381d9a · 2381d9a
1 parent b413ea2
commit 2381d9a
Showing 1 changed file with 12 additions and 0 deletions.
diff --git a/src/aerospike_vector_search/types.py b/src/aerospike_vector_search/types.py
@@ -178,24 +178,36 @@ def __lt__(self, other) -> bool:
         if not isinstance(other, Neighbor):
             return NotImplemented
 
+        if self.distance == other.distance:
+            return self.key.key < other.key.key
+
         return self.distance < other.distance
 
     def __le__(self, other) -> bool:
         if not isinstance(other, Neighbor):
             return NotImplemented
 
+        if self.distance == other.distance:
+            return self.key.key <= other.key.key
+
         return self.distance <= other.distance
 
     def __gt__(self, other) -> bool:
         if not isinstance(other, Neighbor):
             return NotImplemented
 
+        if self.distance == other.distance:
+            return self.key.key > other.key.key
+
         return self.distance > other.distance
 
     def __ge__(self, other) -> bool:
         if not isinstance(other, Neighbor):
             return NotImplemented
 
+        if self.distance == other.distance:
+            return self.key.key >= other.key.key
+
         return self.distance >= other.distance