santit96 commited on
Commit
db761e2
·
1 Parent(s): 86f1c6b
Files changed (2) hide show
  1. wordle_env/const.py +1 -1
  2. wordle_env/state.py +1 -1
wordle_env/const.py CHANGED
@@ -1,4 +1,4 @@
1
  WORDLE_CHARS = 'ABCDEFGHIJKLMNOPQRSTUVWXYZ'
2
  WORDLE_N = 5
3
  REWARD = 10
4
- CHAR_REWARD = 0.1
 
1
  WORDLE_CHARS = 'ABCDEFGHIJKLMNOPQRSTUVWXYZ'
2
  WORDLE_N = 5
3
  REWARD = 10
4
+ CHAR_REWARD = 1
wordle_env/state.py CHANGED
@@ -138,7 +138,7 @@ def update_mask(state: WordleState, word: str, goal_word: str) -> WordleState:
138
  return update_from_mask(state, word, mask)
139
 
140
 
141
- def update(state: WordleState, word: str, goal_word: str) -> Tuple(WordleState, float):
142
  state = state.copy()
143
  reward = 0
144
  state[0] -= 1
 
138
  return update_from_mask(state, word, mask)
139
 
140
 
141
+ def update(state: WordleState, word: str, goal_word: str) -> Tuple[WordleState, float]:
142
  state = state.copy()
143
  reward = 0
144
  state[0] -= 1