Skip to content

Commit ca80acf

Browse files
committed
refactor: Removed magic numbers from terran state
1 parent 9858508 commit ca80acf

File tree

1 file changed

+9
-9
lines changed

1 file changed

+9
-9
lines changed

urnai/sc2/states/terran_state.py

Lines changed: 9 additions & 9 deletions
Original file line numberDiff line numberDiff line change
@@ -26,19 +26,19 @@ def update(self, obs):
2626
units_amount_info = [
2727
raw_units_amount_dict[units.Terran.CommandCenter]
2828
+ raw_units_amount_dict[units.Terran.OrbitalCommand]
29-
+ raw_units_amount_dict[units.Terran.PlanetaryFortress] / 2,
30-
raw_units_amount_dict[units.Terran.SupplyDepot] / 18,
31-
raw_units_amount_dict[units.Terran.Refinery] / 4,
29+
+ raw_units_amount_dict[units.Terran.PlanetaryFortress],
30+
raw_units_amount_dict[units.Terran.SupplyDepot],
31+
raw_units_amount_dict[units.Terran.Refinery],
3232
raw_units_amount_dict[units.Terran.EngineeringBay],
3333
raw_units_amount_dict[units.Terran.Armory],
34-
raw_units_amount_dict[units.Terran.MissileTurret] / 4,
35-
raw_units_amount_dict[units.Terran.SensorTower] / 1,
36-
raw_units_amount_dict[units.Terran.Bunker] / 4,
34+
raw_units_amount_dict[units.Terran.MissileTurret],
35+
raw_units_amount_dict[units.Terran.SensorTower],
36+
raw_units_amount_dict[units.Terran.Bunker],
3737
raw_units_amount_dict[units.Terran.FusionCore],
3838
raw_units_amount_dict[units.Terran.GhostAcademy],
39-
raw_units_amount_dict[units.Terran.Barracks] / 3,
40-
raw_units_amount_dict[units.Terran.Factory] / 2,
41-
raw_units_amount_dict[units.Terran.Starport] / 2,
39+
raw_units_amount_dict[units.Terran.Barracks],
40+
raw_units_amount_dict[units.Terran.Factory],
41+
raw_units_amount_dict[units.Terran.Starport],
4242
]
4343
state = np.squeeze(state)
4444
state = np.append(state, units_amount_info)

0 commit comments

Comments
 (0)