Fix readbit

thatguy11325 · Mar 22, 2024 · 6513fdd · 6513fdd
1 parent cb7c675
commit 6513fdd
Show file tree

Hide file tree

Showing 2 changed files with 2 additions and 2 deletions.
diff --git a/config.yaml b/config.yaml
@@ -7,6 +7,7 @@ debug:
   env:
     headless: False
     stream_wrapper: False
+    init_state: cut
   train:
     device: cpu
     compile: False

diff --git a/pokemonred_puffer/environment.py b/pokemonred_puffer/environment.py
@@ -731,8 +731,7 @@ def read_short(self, addr: str | int) -> int:
 
     def read_bit(self, addr: str | int, bit: int) -> bool:
         # add padding so zero will read '0b100000000' instead of '0b0'
-        # return bin(256 + self.read_m(addr))[-bit - 1] == "1"
-        return bool(int(self.read_m(addr)) & (1 << (7 - bit)))
+        return bool(int(self.read_m(addr)) & (1 << bit))
 
     def read_event_bits(self):
         _, addr = self.pyboy.symbol_lookup("wEventFlags")