Add needed operators to DefaultIntegration

Farama-Foundation · Nov 16, 2023 · 22a52f7 · 22a52f7
1 parent 530a147
commit 22a52f7
Showing 1 changed file with 29 additions and 1 deletion.
diff --git a/retro/data/__init__.py b/retro/data/__init__.py
@@ -50,17 +50,45 @@ def __or__(self, b):
             return False
         return DefaultIntegrations.DEFAULT.value | b
 
+    def __ror__(self, b):
+        try:
+            self._init()
+        except NameError:
+            return False
+        return DefaultIntegrations.DEFAULT.value | b
+
     def __and__(self, b):
         try:
             self._init()
         except NameError:
             return False
         return DefaultIntegrations.DEFAULT.value & b
 
+    def __rand__(self, b):
+        try:
+            self._init()
+        except NameError:
+            return False
+        return DefaultIntegrations.DEFAULT.value & b
+
+    def __lt__(self, b):
+        try:
+            self._init()
+        except NameError:
+            return False
+        return DefaultIntegrations.DEFAULT.value < b
+
+    def __sub__(self, b):
+        try:
+            self._init()
+        except NameError:
+            return False
+        return DefaultIntegrations.DEFAULT.value - b
+
     @classmethod
     def add(cls, extra):
         cls._init()
-        cls.DEFAULT |= extra
+        cls.DEFAULT = cls.DEFAULT | extra
 
     @classmethod
     def reset(cls):