diff --git a/subtrees/dagflow/dagflow/output.py b/subtrees/dagflow/dagflow/output.py index 8147fa035c554b5dda8dcf17f91a781eefbe49bb..5f98ed7e2fce9add4388a08138880171de3c9bb6 100644 --- a/subtrees/dagflow/dagflow/output.py +++ b/subtrees/dagflow/dagflow/output.py @@ -317,6 +317,22 @@ class Output: return True + def seti(self, idx: int, value: float, check_taint: bool = False, force: bool = False) -> bool: + if self.node._frozen and not force: + return False + + tainted = True + if check_taint: + tainted = self._data[udx] != value + + if tainted: + self._data[idx] = value + self.taint_children() + self.node.invalidate_parents() + self.node._tainted = False + + return tainted + def set( self, data: ArrayLike, check_taint: bool = False, force: bool = False ) -> bool: