Merge pull request pybites#8 from pickfire/patch-2

pybites · web-flow · commit 0d5f24077b0a · 2017-12-02T12:44:58.000+01:00
Remove need to convert list to set
diff --git a/039/word_checker.py b/039/word_checker.py
@@ -12,16 +12,8 @@ def load_dictionary():
     return words
 	
 def check_word(dictionary, word):
-    valid_words = []
-    permutations = []
-    perms = list(itertools.permutations(word))
-    for i in perms:
-        permutations.append(''.join(i))
-
-    for i in permutations:
-        if i in dictionary:
-            valid_words.append(i)
-    return set(valid_words)
+    permutations = set(map(''.join, itertools.permutations(word)))
+    return {w for w in permutations if w in dictionary}
 	
 def print_words(words):
     for i in words:
