diff --git a/src/pystencils/backend/ast/iteration.py b/src/pystencils/backend/ast/iteration.py index cc666c72257c5a434c0b312c9e8d8e5ea9dc028f..a3152863a494b7d2b1da2d7388b8ad35bcca0a1c 100644 --- a/src/pystencils/backend/ast/iteration.py +++ b/src/pystencils/backend/ast/iteration.py @@ -10,7 +10,7 @@ def dfs_preorder( Args: node: The tree's root node - filter_pred: Filter predicate; a node is only returned to the caller if `yield_pred(node)` returns True + filter_pred: Filter predicate; a node is only returned to the caller if ``yield_pred(node)`` returns True """ if filter_pred(node): yield node @@ -26,7 +26,7 @@ def dfs_postorder( Args: node: The tree's root node - filter_pred: Filter predicate; a node is only returned to the caller if `yield_pred(node)` returns True + filter_pred: Filter predicate; a node is only returned to the caller if ``yield_pred(node)`` returns True """ for c in node.children: yield from dfs_postorder(c, filter_pred) diff --git a/src/pystencils/backend/ast/util.py b/src/pystencils/backend/ast/util.py index 288097a901e3f11f4a6f12c47799b25ec672151e..27a643054e824bfecd2de1e91eda1e60d27b2e03 100644 --- a/src/pystencils/backend/ast/util.py +++ b/src/pystencils/backend/ast/util.py @@ -20,7 +20,7 @@ def failing_cast(target: type | tuple[type, ...], obj: Any) -> Any: class AstEqWrapper: """Wrapper around AST nodes that computes a hash from the AST's textual representation - and maps the `__eq__` method onto `structurally_equal`. + and maps the ``__eq__`` method onto `structurally_equal <PsAstNode.structurally_equal>`. Useful in dictionaries when the goal is to keep track of subtrees according to their structure, e.g. in elimination of constants or common subexpressions.