diff --git a/sc2/data.py b/sc2/data.py index 7c386052..c5f1605b 100644 --- a/sc2/data.py +++ b/sc2/data.py @@ -43,6 +43,7 @@ Race.Protoss: UnitTypeId.PROBE, Race.Terran: UnitTypeId.SCV, Race.Zerg: UnitTypeId.DRONE, + Race.Random: UnitTypeId.DRONE, } race_townhalls: Dict[Race, Set[UnitTypeId]] = { @@ -84,4 +85,5 @@ Race.Protoss: UnitTypeId.ASSIMILATOR, Race.Terran: UnitTypeId.REFINERY, Race.Zerg: UnitTypeId.EXTRACTOR, + Race.Random: UnitTypeId.EXTRACTOR, }