Fix pyboy sample action to be continuous

UoA-CARES · Feb 27, 2024 · 8f1d83f · 8f1d83f
1 parent 047c24c
commit 8f1d83f
Show file tree

Hide file tree

Showing 2 changed files with 6 additions and 2 deletions.
diff --git a/scripts/envrionments/pyboy/mario/mario_environment.py b/scripts/envrionments/pyboy/mario/mario_environment.py
@@ -29,7 +29,9 @@ def __init__(self, config: GymEnvironmentConfig) -> None:
         ]
 
     def sample_action(self) -> int:
-        return np.random.randint(0, len(self.valid_actions))
+        return np.random.uniform(
+                self.min_action_value, self.max_action_value, size=self.action_num
+            )
 
     def _stats_to_state(self, game_stats: Dict[str, int]) -> List:
         state: List = []

diff --git a/scripts/envrionments/pyboy/pokemon/pokemon_environment.py b/scripts/envrionments/pyboy/pokemon/pokemon_environment.py
@@ -30,7 +30,9 @@ def __init__(self, config: GymEnvironmentConfig) -> None:
         ]
 
     def sample_action(self) -> int:
-        return np.random.randint(0, len(self.valid_actions))
+        return np.random.uniform(
+                self.min_action_value, self.max_action_value, size=self.action_num
+            )
 
     def _stats_to_state(self, game_stats: Dict[str, any]) -> List[any]:
         state: List[any] = []