Skip to content

Commit 21933b0

Browse files
committed
refactor: format result_dict for consistency across outcome functions
1 parent 1f0c38b commit 21933b0

File tree

3 files changed

+27
-3
lines changed

3 files changed

+27
-3
lines changed

src/sc2_datasets/transforms/mmr_vs_result.py

Lines changed: 9 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -29,7 +29,15 @@ def mmr_vs_result(sc2_replay: SC2ReplayData) -> Tuple[torch.Tensor, torch.Tensor
2929
dtype=torch.float,
3030
)
3131

32-
result_dict = {"Loss": 0, "Win": 1, "Victory": 1, "Defeat": 0, "Undecided": -1, "Draw": -1, "Tie": -1}
32+
result_dict = {
33+
"Loss": 0,
34+
"Win": 1,
35+
"Victory": 1,
36+
"Defeat": 0,
37+
"Undecided": -1,
38+
"Draw": -1,
39+
"Tie": -1,
40+
}
3341

3442
# Check if result is "Undecided", "Draw", or "Tie" and return None to skip this replay
3543
skip_results = ["Undecided", "Draw", "Tie"]

src/sc2_datasets/transforms/pytorch/economy_vs_outcome.py

Lines changed: 9 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -59,7 +59,15 @@ def economy_average_vs_outcome(
5959
# Creating feature tensor:
6060
feature_tensor = torch.tensor(feature_list, dtype=torch.float32)
6161

62-
result_dict = {"Loss": 0, "Win": 1, "Victory": 1, "Defeat": 0, "Undecided": -1, "Draw": -1, "Tie": -1}
62+
result_dict = {
63+
"Loss": 0,
64+
"Win": 1,
65+
"Victory": 1,
66+
"Defeat": 0,
67+
"Undecided": -1,
68+
"Draw": -1,
69+
"Tie": -1,
70+
}
6371

6472
# Check if any player's result is "Undecided", "Draw", or "Tie" and return None to skip this replay
6573
skip_results = ["Undecided", "Draw", "Tie"]

src/sc2_datasets/transforms/utils.py

Lines changed: 9 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -235,7 +235,15 @@ def select_outcome_1v1(sc2_replay: SC2ReplayData) -> Dict[str, int]:
235235

236236
player_outcome = {"1": 0, "2": 0}
237237

238-
result_dict = {"Loss": 0, "Win": 1, "Victory": 1, "Defeat": 0, "Undecided": -1, "Draw": -1, "Tie": -1}
238+
result_dict = {
239+
"Loss": 0,
240+
"Win": 1,
241+
"Victory": 1,
242+
"Defeat": 0,
243+
"Undecided": -1,
244+
"Draw": -1,
245+
"Tie": -1,
246+
}
239247

240248
# Check if any player has an "Undecided", "Draw", or "Tie" result and return None to indicate skipping
241249
skip_results = ["Undecided", "Draw", "Tie"]

0 commit comments

Comments
 (0)