Fix enum names changed from new ID generation script

Author: Reynald Pader

examples/zerg_rush_build_order.py

@@ -43,15 +43,15 @@ async def on_step(self, iteration):
         if self.vespene >= 100:
             sp = self.units(UnitTypeId.SPAWNINGPOOL).ready
             if sp.exists and self.minerals >= 100 and not self.mboost_started:
-                await self.do(sp.first(AbilityId.ZERGLINGMOVEMENTSPEED))
+                await self.do(sp.first(AbilityId.RESEARCH_ZERGLINGMETABOLICBOOST))
                 self.mboost_started = True
 
         await self.build_order.execute_build()
 
         for queen in self.units(UnitTypeId.QUEEN).idle:
             if queen.energy >= 25:  # Hard coded, since this is not (yet) available
                 hatchery = self.townhalls.closest_to(queen.position.to2)
-                await self.do(queen(AbilityId.INJECTLARVA, hatchery))
+                await self.do(queen(AbilityId.EFFECT_INJECTLARVA, hatchery))
 
         if (self.units(UnitTypeId.ROACH).amount >= 7 and self.units(UnitTypeId.ZERGLING).amount >= 10) or self.attack:
             self.attack = True

sc2/bot_ai.py

@@ -148,7 +148,7 @@ async def distribute_workers(self):
             for x in range(0, deficit):
                 if worker_pool:
                     w = worker_pool.pop()
-                    if len(w.orders) == 1 and w.orders[0].ability.id in [AbilityId.HARVESTRETURN]:
+                    if len(w.orders) == 1 and w.orders[0].ability.id in [AbilityId.HARVEST_RETURN]:
                         await self.do(w.move(g))
                         await self.do(w.return_resource(queue=True))
                     else:
@@ -163,7 +163,7 @@ async def distribute_workers(self):
                 if worker_pool:
                     w = worker_pool.pop()
                     mf = self.state.mineral_field.closest_to(townhall)
-                    if len(w.orders) == 1 and w.orders[0].ability.id in [AbilityId.HARVESTRETURN]:
+                    if len(w.orders) == 1 and w.orders[0].ability.id in [AbilityId.HARVEST_RETURN]:
                         await self.do(w.move(townhall))
                         await self.do(w.return_resource(queue=True))
                         await self.do(w.gather(mf, queue=True))

sc2/unit.py

@@ -197,7 +197,7 @@ def gather(self, *args, **kwargs):
         return self(AbilityId.HARVEST_GATHER, *args, **kwargs)
 
     def return_resource(self, *args, **kwargs):
-        return self(AbilityId.HARVESTRETURN, *args, **kwargs)
+        return self(AbilityId.HARVEST_RETURN, *args, **kwargs)
 
     def move(self, *args, **kwargs):
         return self(AbilityId.MOVE, *args, **kwargs)