Skip to content

Commit 709df53

Browse files
committed
Patches a couple processors to better handle replays with computer players
1 parent c734c8e commit 709df53

File tree

1 file changed

+8
-2
lines changed

1 file changed

+8
-2
lines changed

sc2reader/processors.py

Lines changed: 8 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -30,7 +30,10 @@ class PeopleProcessor(Processor):
3030
def process(self, replay):
3131
obs_players = list(replay.player_names)
3232
for player in replay.players:
33-
obs_players.remove(player.name)
33+
try:
34+
obs_players.remove(player.name)
35+
except ValueError:
36+
pass #Must be a computer player!!
3437

3538
for pid,name in enumerate(obs_players):
3639
replay.observers.append(Observer(pid+len(replay.players)+1,name,replay))
@@ -155,7 +158,10 @@ def process(self, replay):
155158

156159
# Average the APM for actual players
157160
for player in replay.players:
158-
player.avg_apm /= player.events[-1].second/60.0
161+
if player.type != "Computer":
162+
player.avg_apm /= player.events[-1].second/60.0
163+
else:
164+
player.avg_amp = 0
159165

160166
return replay
161167

0 commit comments

Comments
 (0)