From 206030ecf7d45b0b8ef4e4ab105302088a0401b3 Mon Sep 17 00:00:00 2001 From: Erez Sh Date: Sat, 27 Mar 2021 16:16:57 -0600 Subject: [PATCH] Docs: Added docstring to Tree.scan_values() --- lark/tree.py | 7 +++++++ 1 file changed, 7 insertions(+) diff --git a/lark/tree.py b/lark/tree.py index 9d95015..bee53cf 100644 --- a/lark/tree.py +++ b/lark/tree.py @@ -108,6 +108,13 @@ class Tree(object): self.children[i:i+1] = kid.children def scan_values(self, pred): + """Return all values in the tree that evaluate pred(value) as true. + + This can be used to find all the tokens in the tree. + + Example: + >>> all_tokens = tree.scan_values(lambda v: isinstance(v, Token)) + """ for c in self.children: if isinstance(c, Tree): for t in c.scan_values(pred):