Add equality and order overloads for BoolWithConfidence

1 files changed, 21 insertions, 0 deletions

diff --git a/python/types.py b/python/types.py
index 67ad8438..2a141846 100644
--- a/python/types.py
+++ b/python/types.py
@@ -369,6 +369,27 @@ class BoolWithConfidence:
 	value:bool
 	confidence:int=core.max_confidence
 
+	def __eq__(self, other):
+		if not isinstance(other, self.__class__):
+			return self.value == bool(other)
+		else:
+			return (self.value, self.confidence) == (other.value, other.confidence)
+
+	def __ne__(self, other):
+		return not (self == other)
+
+	def __gt__(self, other):
+		return self.confidence > other.confidence
+
+	def __le__(self, other):
+		return self.confidence <= other.confidence
+
+	def __ge__(self, other):
+		return self.confidence >= other.confidence
+
+	def __lt__(self, other):
+		return self.confidence < other.confidence
+
 	def __bool__(self):
 		return self.value