diff --git a/gym_csgo/spaces/obs.py b/gym_csgo/spaces/obs.py index 105f780..d1e88fd 100644 --- a/gym_csgo/spaces/obs.py +++ b/gym_csgo/spaces/obs.py @@ -19,7 +19,7 @@ def make_weapon_obs_space(): # Type of the weapon 'type': Enum(*WEAPON_TYPES, 'none'), # State of weapon - 'state': Enum('active', 'holstered', 'none'), + 'state': Enum('active', 'holstered', 'reloading', 'none'), # Current amount of ammo in the clip 'ammo_clip': Box(low=0, high=np.inf, shape=(1,)), # Maximum amount of ammo the clip for the weapon can hold diff --git a/gym_csgo/specs/weapon.py b/gym_csgo/specs/weapon.py index 0430f31..2ee5fd6 100644 --- a/gym_csgo/specs/weapon.py +++ b/gym_csgo/specs/weapon.py @@ -17,7 +17,8 @@ # List of known CSGO weapon types WEAPON_TYPES = [ - 'Pistol', 'Knife', 'Rifle', 'SniperRifle', 'Submachine Gun', 'C4', 'Grenade' + 'Pistol', 'Knife', 'Rifle', 'SniperRifle', 'Submachine Gun', 'C4', + 'Grenade', 'Shotgun', 'Machine Gun', 'StackableItem' ] # Weapon slots to consider