File tree Expand file tree Collapse file tree
Expand file tree Collapse file tree Original file line number Diff line number Diff line change @@ -476,10 +476,16 @@ def _prepare_start(
476476 self .realtime : bool = realtime
477477 self .base_build : int = base_build
478478
479- self .race : Race = Race (self .game_info .player_races [self .player_id ])
479+ if self .player_id > 0 :
480+ self .race : Race = Race (self .game_info .player_races [self .player_id ])
480481
481- if len (self .game_info .player_races ) == 2 :
482- self .enemy_race : Race = Race (self .game_info .player_races [3 - self .player_id ])
482+ if len (self .game_info .player_races ) == 2 :
483+ self .enemy_race : Race = Race (self .game_info .player_races [3 - self .player_id ])
484+ else :
485+ # Observer
486+ self .race : Race = Race .Random
487+
488+ self ._distances_override_functions (self .distance_calculation_method )
483489
484490 self ._distances_override_functions (self .distance_calculation_method )
485491
Original file line number Diff line number Diff line change 66
77from __future__ import annotations
88
9- from typing import TYPE_CHECKING
10-
119from sc2 .bot_ai_internal import BotAIInternal
1210from sc2 .data import Alert , Result
13- from sc2 .game_data import GameData
1411from sc2 .ids .ability_id import AbilityId
1512from sc2 .ids .upgrade_id import UpgradeId
1613from sc2 .position import Point2
1714from sc2 .unit import Unit
1815from sc2 .units import Units
1916
20- if TYPE_CHECKING :
21- from sc2 .client import Client
22- from sc2 .game_info import GameInfo
23-
2417
2518class ObserverAI (BotAIInternal ):
2619 """Base class for bots."""
@@ -36,21 +29,6 @@ def time_formatted(self) -> str:
3629 t = self .time
3730 return f"{ int (t // 60 ):02} :{ int (t % 60 ):02} "
3831
39- @property
40- def game_info (self ) -> GameInfo :
41- """See game_info.py"""
42- return self ._game_info
43-
44- @property
45- def game_data (self ) -> GameData :
46- """See game_data.py"""
47- return self ._game_data
48-
49- @property
50- def client (self ) -> Client :
51- """See client.py"""
52- return self ._client
53-
5432 def alert (self , alert_code : Alert ) -> bool :
5533 """
5634 Check if alert is triggered in the current step.
You can’t perform that action at this time.
0 commit comments