Merge pull request #303 from XpressAI/wmeddie-none-checks

Add IsNone and IsNotNone components
XpressAI · Mar 7, 2024 · ef856da · ef856da
2 parents 1bcebd6 + 40cc36f
commit ef856da
Showing 1 changed file with 21 additions and 0 deletions.
diff --git a/xai_components/xai_controlflow/branches.py b/xai_components/xai_controlflow/branches.py
@@ -114,3 +114,24 @@ def __init__(self):
     def execute(self, ctx) -> None:
         ctx[self.name.value] = self.value.value
         self.ref.set_fn(lambda: ctx[self.name.value])
+
+
+@xai_component
+class IsNone(Component):
+    a: InArg[any]
+
+    out: OutArg[bool]
+
+    def execute(self, ctx) -> None:
+        self.out.value = self.a.value is None
+
+
+@xai_component
+class IsNotNone(Component):
+    a: InArg[any]
+
+    out: OutArg[bool]
+
+    def execute(self, ctx) -> None:
+        self.out.value = self.a.value is not None
+