Add sound option to initial config

msosav · Jan 7, 2025 · 185c6fe · 185c6fe
1 parent cd4093c
commit 185c6fe
Show file tree

Hide file tree

Showing 2 changed files with 2 additions and 1 deletion.
diff --git a/config/gym.py b/config/gym.py
@@ -15,7 +15,7 @@ def __init__(self, config: dict, debug=False):
 
         assert self.rom_path is not None, "ROM path is required"
 
-        self.pyboy = PyBoy(self.rom_path)
+        self.pyboy = PyBoy(self.rom_path, sound=config["game_with_sound"])
 
         self._fitness = 0
         self._previous_fitness = 0

diff --git a/main.py b/main.py
@@ -12,6 +12,7 @@
         "rom_path": "roms/ZeldaLinksAwakening.gb",
         "checkpoint_dir": "checkpoints/",
         "log_dir": "logs/",
+        "game_with_sound": True,
     }
 
     callback = CheckpointAndLoggingCallback(