Fix enum names changed from new ID generation script

Author: Reynald Pader

examples/zerg_rush_build_order.py

@@ -43,7 +43,7 @@ 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()

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)