diff --git a/models/team.py b/models/team.py index 3af4573..8b1c78b 100644 --- a/models/team.py +++ b/models/team.py @@ -1,16 +1,13 @@ from models.player import Player class Team: - def __init__(self): + def __init__(self, is_ct=False): self.players = [] - self.is_ct = False + self.is_ct = is_ct self.score = 0 def add_player(self, player: Player): self.players.append(player) def remove_player(self, player: Player): - self.players.remove(player) - - def set_ct(self, is_ct: bool): - self.is_ct = is_ct \ No newline at end of file + self.players.remove(player) \ No newline at end of file