Shorten Symbol/Choice.referenced()

Use a set comprehension.

1 files changed, 2 insertions, 10 deletions

diff --git a/kconfiglib.py b/kconfiglib.py
index 6ead7d7..2cd1a29 100644
--- a/kconfiglib.py
+++ b/kconfiglib.py
@@ -4299,11 +4299,7 @@ class Symbol(object):
         """
         See the class documentation.
         """
-        res = set()
-        for node in self.nodes:
-            res |= node.referenced
-
-        return res
+        return {item for node in self.nodes for item in node.referenced}
 
     def __repr__(self):
         """
@@ -4894,11 +4890,7 @@ class Choice(object):
         """
         See the class documentation.
         """
-        res = set()
-        for node in self.nodes:
-            res |= node.referenced
-
-        return res
+        return {item for node in self.nodes for item in node.referenced}
 
     def __repr__(self):
         """