Added a few flake8 plugin

1 files changed, 5 insertions, 5 deletions

diff --git a/src/suite/suite.py b/src/suite/suite.py
index f3d58b1..836cac0 100644
--- a/src/suite/suite.py
+++ b/src/suite/suite.py
@@ -6,7 +6,7 @@
 #    By: charles <charles.cabergs@gmail.com>        +#+  +:+       +#+         #
 #                                                 +#+#+#+#+#+   +#+            #
 #    Created: 2020/07/15 18:24:29 by charles           #+#    #+#              #
-#    Updated: 2021/01/31 04:40:22 by charles          ###   ########.fr        #
+#    Updated: 2021/02/04 16:13:08 by charles          ###   ########.fr        #
 #                                                                              #
 # ############################################################################ #
 
@@ -28,7 +28,7 @@ class Suite:
                 break
 
     @classmethod
-    def setup(cls, asked_names: List[str]):
+    def setup(cls, asked_names: List[str]) -> None:
         """ Remove not asked suite from available suites
             Tries to autocomplete the asked names
         """
@@ -49,7 +49,7 @@ class Suite:
                        or n.startswith(name)]
             if len(matches) == 1:
                 names.append(matches[0])
-            elif len(matches) != 0 and all([n.startswith(name) for n in matches]):
+            elif len(matches) != 0 and all(n.startswith(name) for n in matches):
                 names.extend(matches)
             elif len(matches) > 2:
                 print(("Ambiguous name `{}` match the following suites\n\t{}\n"
@@ -64,7 +64,7 @@ class Suite:
 
         cls.available = list(set(
             [s for s in cls.available if s.name in names]
-            + [s for s in cls.available if any([g for g in s.groups if g in names])]
+            + [s for s in cls.available if any(g for g in s.groups if g in names)]
         ))
         cls.available.sort(key=lambda s: s.name)
         for s in cls.available:
@@ -79,7 +79,7 @@ class Suite:
     @classmethod
     def list(cls):
         print("Groups:")
-        print("\n".join(set([" - " + ', '.join(s.groups) for s in Suite.available])))
+        print("\n".join({" - " + ', '.join(s.groups) for s in Suite.available}))
         print("The available suites are:")
         max_name_width = max(len(s.name) for s in Suite.available) + 5
         lines = [