Fix PyCQA#108

glennmatthews · Feb 19, 2015 · abb3d44 · abb3d44
1 parent 286f4ff
commit abb3d44
Showing 1 changed file with 2 additions and 3 deletions.
diff --git a/pep257.py b/pep257.py
@@ -474,9 +474,8 @@ def collect(names, match=lambda name: True, match_dir=lambda name: True):
     for name in names:  # map(expanduser, names):
         if os.path.isdir(name):
             for root, dirs, filenames in os.walk(name):
-                for dir in dirs:
-                    if not match_dir(dir):
-                        dirs.remove(dir)  # do not visit those dirs
+                # Skip any dirs that do not match match_dir
+                dirs[:] = [dir for dir in dirs if match_dir(dir)]
                 for filename in filenames:
                     if match(filename):
                         yield os.path.join(root, filename)