Switch to set comprehension

willkg · willkg · commit 817dff46d239 · 2017-11-06T16:18:02.000-05:00
diff --git a/html5lib/_trie/_base.py b/html5lib/_trie/_base.py
@@ -13,7 +13,7 @@ def keys(self, prefix=None):
         if prefix is None:
             return set(keys)
 
-        return set([x for x in keys if x.startswith(prefix)])
+        return {x for x in keys if x.startswith(prefix)}
 
     def has_keys_with_prefix(self, prefix):
         for key in self.keys():
