Add the units and actions from the 4.10.1 balance patch.

PiperOrigin-RevId: 267670019

pysc2/lib/actions.py

@@ -999,6 +999,7 @@ def __eq__(self, other):
     Function.ability(570, "Research_CycloneLockOnDamage_quick", cmd_quick, 769),
     Function.ability(540, "Research_CycloneRapidFireLaunchers_quick", cmd_quick, 768),
     Function.ability(363, "Research_DrillingClaws_quick", cmd_quick, 764),
+    Function.ability(572, "Research_EnhancedShockwaves_quick", cmd_quick, 822),
     Function.ability(364, "Research_ExtendedThermalLance_quick", cmd_quick, 1097),
     Function.ability(365, "Research_GlialRegeneration_quick", cmd_quick, 216),
     Function.ability(366, "Research_GraviticBooster_quick", cmd_quick, 1093),
@@ -1575,6 +1576,7 @@ def __eq__(self, other):
     Function.raw_ability(554, "Research_CycloneLockOnDamage_quick", raw_cmd, 769),
     Function.raw_ability(435, "Research_CycloneRapidFireLaunchers_quick", raw_cmd, 768),
     Function.raw_ability(436, "Research_DrillingClaws_quick", raw_cmd, 764),
+    Function.raw_ability(563, "Research_EnhancedShockwaves_quick", raw_cmd, 822),
     Function.raw_ability(69, "Research_ExtendedThermalLance_quick", raw_cmd, 1097),
     Function.raw_ability(437, "Research_GlialRegeneration_quick", raw_cmd, 216),
     Function.raw_ability(67, "Research_GraviticBooster_quick", raw_cmd, 1093),

pysc2/lib/static_data.py

@@ -70,30 +70,30 @@ def general_abilities(self):
     522, 524, 554, 556, 558, 560, 561, 562, 563, 591, 594, 595, 596, 597, 614,
     620, 621, 622, 623, 624, 626, 650, 651, 652, 653, 654, 655, 656, 657, 658,
     710, 730, 731, 732, 761, 764, 766, 768, 769, 790, 793, 799, 803, 804, 805,
-    820, 855, 856, 857, 861, 862, 863, 864, 865, 866, 880, 881, 882, 883, 884,
-    885, 886, 887, 889, 890, 891, 892, 893, 894, 895, 911, 913, 914, 916, 917,
-    919, 920, 921, 922, 946, 948, 950, 954, 955, 976, 977, 978, 979, 994, 1006,
-    1036, 1038, 1039, 1042, 1062, 1063, 1064, 1065, 1066, 1067, 1068, 1069,
-    1070, 1093, 1094, 1097, 1126, 1152, 1154, 1155, 1156, 1157, 1158, 1159,
-    1160, 1161, 1162, 1163, 1165, 1166, 1167, 1183, 1184, 1186, 1187, 1188,
-    1189, 1190, 1191, 1192, 1193, 1194, 1216, 1217, 1218, 1219, 1220, 1221,
-    1223, 1225, 1252, 1253, 1282, 1283, 1312, 1313, 1314, 1315, 1316, 1317,
-    1342, 1343, 1344, 1345, 1346, 1348, 1351, 1352, 1353, 1354, 1356, 1372,
-    1373, 1374, 1376, 1378, 1380, 1382, 1384, 1386, 1388, 1390, 1392, 1394,
-    1396, 1406, 1408, 1409, 1413, 1414, 1416, 1417, 1418, 1419, 1433, 1435,
-    1437, 1438, 1440, 1442, 1444, 1446, 1448, 1449, 1450, 1451, 1454, 1455,
-    1482, 1512, 1514, 1516, 1517, 1518, 1520, 1522, 1524, 1526, 1528, 1530,
-    1532, 1562, 1563, 1564, 1565, 1566, 1567, 1568, 1592, 1593, 1594, 1622,
-    1623, 1628, 1632, 1664, 1682, 1683, 1684, 1691, 1692, 1693, 1694, 1725,
-    1727, 1729, 1730, 1731, 1732, 1733, 1763, 1764, 1766, 1768, 1819, 1825,
-    1831, 1832, 1833, 1834, 1847, 1848, 1853, 1974, 1978, 1998, 2014, 2016,
-    2048, 2057, 2063, 2067, 2073, 2081, 2082, 2095, 2097, 2099, 2108, 2110,
-    2112, 2113, 2114, 2116, 2146, 2162, 2244, 2324, 2328, 2330, 2331, 2332,
-    2333, 2338, 2340, 2342, 2346, 2350, 2354, 2358, 2362, 2364, 2365, 2368,
-    2370, 2371, 2373, 2375, 2376, 2387, 2389, 2391, 2393, 2505, 2535, 2542,
-    2544, 2550, 2552, 2558, 2560, 2588, 2594, 2596, 2700, 2704, 2708, 2709,
-    2714, 2720, 3707, 3709, 3739, 3741, 3743, 3745, 3747, 3749, 3751, 3753,
-    3755, 3757, 3765, 3771, 3776, 3777, 3778, 3783,
+    820, 822, 855, 856, 857, 861, 862, 863, 864, 865, 866, 880, 881, 882, 883,
+    884, 885, 886, 887, 889, 890, 891, 892, 893, 894, 895, 911, 913, 914, 916,
+    917, 919, 920, 921, 922, 946, 948, 950, 954, 955, 976, 977, 978, 979, 994,
+    1006, 1036, 1038, 1039, 1042, 1062, 1063, 1064, 1065, 1066, 1067, 1068,
+    1069, 1070, 1093, 1094, 1097, 1126, 1152, 1154, 1155, 1156, 1157, 1158,
+    1159, 1160, 1161, 1162, 1163, 1165, 1166, 1167, 1183, 1184, 1186, 1187,
+    1188, 1189, 1190, 1191, 1192, 1193, 1194, 1216, 1217, 1218, 1219, 1220,
+    1221, 1223, 1225, 1252, 1253, 1282, 1283, 1312, 1313, 1314, 1315, 1316,
+    1317, 1342, 1343, 1344, 1345, 1346, 1348, 1351, 1352, 1353, 1354, 1356,
+    1372, 1373, 1374, 1376, 1378, 1380, 1382, 1384, 1386, 1388, 1390, 1392,
+    1394, 1396, 1406, 1408, 1409, 1413, 1414, 1416, 1417, 1418, 1419, 1433,
+    1435, 1437, 1438, 1440, 1442, 1444, 1446, 1448, 1449, 1450, 1451, 1454,
+    1455, 1482, 1512, 1514, 1516, 1517, 1518, 1520, 1522, 1524, 1526, 1528,
+    1530, 1532, 1562, 1563, 1564, 1565, 1566, 1567, 1568, 1592, 1593, 1594,
+    1622, 1623, 1628, 1632, 1664, 1682, 1683, 1684, 1691, 1692, 1693, 1694,
+    1725, 1727, 1729, 1730, 1731, 1732, 1733, 1763, 1764, 1766, 1768, 1819,
+    1825, 1831, 1832, 1833, 1834, 1847, 1848, 1853, 1974, 1978, 1998, 2014,
+    2016, 2048, 2057, 2063, 2067, 2073, 2081, 2082, 2095, 2097, 2099, 2108,
+    2110, 2112, 2113, 2114, 2116, 2146, 2162, 2244, 2324, 2328, 2330, 2331,
+    2332, 2333, 2338, 2340, 2342, 2346, 2350, 2354, 2358, 2362, 2364, 2365,
+    2368, 2370, 2371, 2373, 2375, 2376, 2387, 2389, 2391, 2393, 2505, 2535,
+    2542, 2544, 2550, 2552, 2558, 2560, 2588, 2594, 2596, 2700, 2704, 2708,
+    2709, 2714, 2720, 3707, 3709, 3739, 3741, 3743, 3745, 3747, 3749, 3751,
+    3753, 3755, 3757, 3765, 3771, 3776, 3777, 3778, 3783,
 ]
 
 # List of known unit types. It is generated by parsing replays and from:
@@ -111,10 +111,10 @@ def general_abilities(self):
     350, 364, 365, 371, 372, 373, 376, 377, 472, 473, 474, 475, 483, 484, 485,
     486, 487, 488, 489, 490, 493, 494, 495, 496, 498, 499, 500, 501, 502, 503,
     504, 517, 518, 559, 560, 561, 562, 563, 564, 588, 589, 590, 591, 608, 609,
-    610, 612, 628, 630, 638, 639, 640, 641, 642, 643, 648, 649, 651, 661, 662,
-    663, 664, 665, 666, 687, 688, 689, 690, 691, 692, 693, 694, 732, 733, 734,
-    796, 797, 801, 824, 830, 877, 880, 881, 884, 885, 886, 887, 892, 893, 894,
-    1904, 1908, 1910, 1911, 1912, 1913, 1955, 1956, 1957, 1958, 1960, 1961
+    610, 612, 628, 629, 630, 638, 639, 640, 641, 642, 643, 648, 649, 651, 661,
+    662, 663, 664, 665, 666, 687, 688, 689, 690, 691, 692, 693, 694, 732, 733,
+    734, 796, 797, 801, 824, 830, 877, 880, 881, 884, 885, 886, 887, 892, 893,
+    894, 1904, 1908, 1910, 1911, 1912, 1913, 1955, 1956, 1957, 1958, 1960, 1961,
 ]
 
 # List of used buffs found by parsing replays.
@@ -130,5 +130,5 @@ def general_abilities(self):
     31, 32, 36, 37, 38, 39, 40, 41, 42, 43, 44, 45, 46, 47, 48, 49, 50, 52, 53,
     54, 55, 56, 57, 58, 59, 60, 61, 62, 64, 65, 66, 68, 69, 70, 71, 72, 73, 74,
     75, 76, 78, 79, 80, 81, 82, 83, 84, 86, 87, 88, 99, 101, 116, 117, 118, 122,
-    130, 134, 135, 136, 139, 140, 141, 144, 289, 291, 293,
+    130, 134, 135, 136, 139, 140, 141, 144, 289, 291, 293, 296,
 ]

pysc2/lib/units.py

@@ -51,6 +51,7 @@ class Neutral(enum.IntEnum):
   DebrisRampRight = 487
   DestructibleBillboardTall = 350
   DestructibleCityDebris4x4 = 628
+  DestructibleCityDebris6x6 = 629
   DestructibleCityDebrisHugeDiagonalBLUR = 630
   DestructibleDebris4x4 = 364
   DestructibleDebris6x6 = 365

pysc2/lib/upgrades.py

@@ -39,6 +39,7 @@ class Upgrades(enum.IntEnum):
   CorvidReactor = 22
   CycloneRapidFireLaunchers = 291
   DrillingClaws = 122
+  EnhancedShockwaves = 296
   ExtendedThermalLance = 50
   GlialReconstitution = 2
   GraviticBooster = 48