Add missing weapon types and states

iksnagreb · Oct 7, 2021 · f5e7c95 · f5e7c95
1 parent 8a4c607
commit f5e7c95
Show file tree

Hide file tree

Showing 2 changed files with 3 additions and 2 deletions.
diff --git 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
@@ -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