From 3b356d21407f8f31c95b9d2f5d852d01bfc08546 Mon Sep 17 00:00:00 2001
From: Frederik Hennig <frederik.hennig@fau.de>
Date: Tue, 12 Dec 2023 10:26:30 +0100
Subject: [PATCH] bugfix to children property in PsBlock
---
pystencils/nbackend/ast/nodes.py | 9 +++++++--
1 file changed, 7 insertions(+), 2 deletions(-)
diff --git a/pystencils/nbackend/ast/nodes.py b/pystencils/nbackend/ast/nodes.py
index 09774f903..16a7bd299 100644
--- a/pystencils/nbackend/ast/nodes.py
+++ b/pystencils/nbackend/ast/nodes.py
@@ -38,8 +38,13 @@ class PsAstNode(ABC):
class PsBlock(PsAstNode):
- @PsAstNode.children.setter
- def children(self, cs):
+
+ @property
+ def children(self) -> Generator[PsAstNode, None, None]:
+ yield from self._children # need to override entire property to override the setter
+
+ @children.setter
+ def children(self, cs: Sequence[PsAstNode]):
self._children = cs
--
GitLab