Added C++ agent

[progcomp2013.git] / qchess / src / piece.py

diff --git a/qchess/src/piece.py b/qchess/src/piece.py
index 81fbb56..a72937c 100644 (file)
--- a/qchess/src/piece.py
+++ b/qchess/src/piece.py
@@ -75,7 +75,7 @@ class Piece():
        
        # Collapses the wave function!          
        def select(self):
-               if self.current_type == "unknown":
+               if self.current_type == "unknown" or not self.choice in [0,1]:
                        self.choice = random.randint(0,1)
                        if self.types[self.choice][0] == '?':
                                self.types[self.choice] = self.types[self.choice][1:]