Skip to content

Commit 78120cf

Browse files
committed
fix: return None for skipped replays in economy_average_vs_outcome function
1 parent 21933b0 commit 78120cf

File tree

1 file changed

+1
-1
lines changed

1 file changed

+1
-1
lines changed

src/sc2_datasets/transforms/pytorch/economy_vs_outcome.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -73,7 +73,7 @@ def economy_average_vs_outcome(
7373
skip_results = ["Undecided", "Draw", "Tie"]
7474
if any(
7575
player_desc.toon_player_info.result in skip_results
76-
for player_desc in sc2_replay.toonPlayerDescMap.values()
76+
for player_desc in sc2_replay.toonPlayerDescMap
7777
):
7878
return None, None
7979

0 commit comments

Comments
 (0)