feat: add support for "Undecided" match results in outcome processing

Author: scmisa

src/sc2_datasets/transforms/mmr_vs_result.py

@@ -29,7 +29,12 @@ def mmr_vs_result(sc2_replay: SC2ReplayData) -> Tuple[torch.Tensor, torch.Tensor
         dtype=torch.float,
     )
 
-    result_dict = {"Loss": 0, "Win": 1, "Victory": 1, "Defeat": 0}
+    result_dict = {"Loss": 0, "Win": 1, "Victory": 1, "Defeat": 0, "Undecided": -1}
+    
+    # Check if result is "Undecided" and return None to skip this replay
+    if sc2_replay.toonPlayerDescMap[0].toon_player_info.result == "Undecided":
+        return None, None
+    
     label_tensor = torch.tensor(
         result_dict[sc2_replay.toonPlayerDescMap[0].toon_player_info.result],
         dtype=torch.int8,

src/sc2_datasets/transforms/pytorch/economy_vs_outcome.py

@@ -59,7 +59,12 @@ def economy_average_vs_outcome(
     # Creating feature tensor:
     feature_tensor = torch.tensor(feature_list, dtype=torch.float32)
 
-    result_dict = {"Loss": 0, "Win": 1, "Victory": 1, "Defeat": 0}
+    result_dict = {"Loss": 0, "Win": 1, "Victory": 1, "Defeat": 0, "Undecided": -1}
+    
+    # Check if result is "Undecided" and return None to skip this replay
+    if sc2_replay.toonPlayerDescMap[0].toon_player_info.result == "Undecided":
+        return None, None
+        
     target = result_dict[sc2_replay.toonPlayerDescMap[0].toon_player_info.result]
 
     return feature_tensor, target

src/sc2_datasets/transforms/utils.py

@@ -235,7 +235,13 @@ def select_outcome_1v1(sc2_replay: SC2ReplayData) -> Dict[str, int]:
 
     player_outcome = {"1": 0, "2": 0}
 
-    result_dict = {"Loss": 0, "Win": 1, "Victory": 1, "Defeat": 0}
+    result_dict = {"Loss": 0, "Win": 1, "Victory": 1, "Defeat": 0, "Undecided": -1}
+    
+    # Check if any player has an "Undecided" result and return None to indicate skipping
+    for toon_desc_map in sc2_replay.toonPlayerDescMap:
+        if toon_desc_map.toon_player_info.result == "Undecided":
+            return None
+            
     for toon_desc_map in sc2_replay.toonPlayerDescMap:
         result = result_dict[toon_desc_map.toon_player_info.result]
         player_outcome[toon_desc_map.toon_player_info.playerID] = result