From 284a29d7aff5df8f3877d24abb05f37915036ac4 Mon Sep 17 00:00:00 2001
From: Ethan Shapiro <46407744+Ethan-Shapiro@users.noreply.github.com>
Date: Wed, 20 Mar 2024 19:53:50 -0700
Subject: [PATCH] f

---
 eric_wordle/ai.py | 4 ++--
 1 file changed, 2 insertions(+), 2 deletions(-)

diff --git a/eric_wordle/ai.py b/eric_wordle/ai.py
index 7af906f..1ab280c 100644
--- a/eric_wordle/ai.py
+++ b/eric_wordle/ai.py
@@ -56,6 +56,7 @@ class AI:
             # get emulated results
             results = results_callback(word)
             if self.use_q_model:
+                self.q_env.set_state(self.q_env_state)
                 # step the q_env to match the guess we just made
                 for i in range(len(word)):
                     char = word[i]
@@ -141,9 +142,9 @@ class AI:
         best_word = None
         for word, _ in self.best_words:
             # reset the state back to before we guessed a word
-            self.q_env.set_state(self.q_env_state)
             if pattern.match(word) and False not in [e in word for e in self.possible_letters]:
                 if self.use_q_model:
+                    self.q_env.set_state(self.q_env_state)
                     # Use policy to grade word
                     # get the state and action pairs
                     curr_qval = 0
@@ -160,7 +161,6 @@ class AI:
                 else:
                     # otherwise return the word from eric heuristic
                     return word
-        self.q_env.set_state(self.q_env_state)
         return best_word
 
     def get_vocab(self, vocab_file):