Add some comments

wordle_env/state.py

@@ -164,6 +164,8 @@ def update(state: WordleState, word: str, goal_word: str) -> Tuple[WordleState,
 
 
 def _set_if_cero(state, offset, value):
+    # set offset character with value at all positions
+    # but only if it didnt have a value before
     for char_idx in range(0, WORDLE_N * 3, 3):
         char_offset = offset + char_idx
         if tuple(state[char_offset: char_offset + 3]) == (0, 0, 0):
@@ -181,8 +183,10 @@ def _set_yes(state, offset, char_int, char_pos):
 
 
 def _set_no(state, offset, char_pos):
+    # Set offset character = no at char_pos position
     state[offset + 3 * char_pos:offset + 3 * char_pos + 3] = [1, 0, 0]
 
 
 def _set_all_no(state, offset):
+    # Set offset character = no at all positions
     state[offset:offset + 3 * WORDLE_N] = [1, 0, 0] * WORDLE_N