Tweak SquareSet constructor

niklasf · Jul 11, 2018 · 3698bf5 · 3698bf5
1 parent 26c0c15
commit 3698bf5
Showing 1 changed file with 6 additions and 2 deletions.
diff --git a/chess/__init__.py b/chess/__init__.py
@@ -3370,10 +3370,14 @@ class SquareSet(collections.abc.MutableSet):
     def __init__(self, squares=BB_EMPTY):
         try:
             self.mask = int(squares) & BB_ALL
+            return
         except TypeError:
             self.mask = 0
-            for square in squares:
-                self.add(square)
+
+        # Try squares as an iterable. Not under except clause for nicer
+        # backtraces.
+        for square in squares:
+            self.add(square)
 
     # Set