diff --git a/chesscog/recognition/recognition.py b/chesscog/recognition/recognition.py index 457a7dc..f00ccda 100644 --- a/chesscog/recognition/recognition.py +++ b/chesscog/recognition/recognition.py @@ -108,7 +108,7 @@ def _classify_pieces(self, img: np.ndarray, turn: chess.Color, corners: np.ndarr pieces = self._pieces_model(piece_imgs) pieces = pieces.argmax(axis=-1).cpu().numpy() pieces = self._piece_classes[pieces] - all_pieces = np.full(len(self._squares), None, dtype=np.object) + all_pieces = np.full(len(self._squares), None, dtype=object) all_pieces[occupancy] = pieces return all_pieces