diff --git a/pokemonred_puffer/cleanrl_puffer.py b/pokemonred_puffer/cleanrl_puffer.py index 6746b42..4306794 100644 --- a/pokemonred_puffer/cleanrl_puffer.py +++ b/pokemonred_puffer/cleanrl_puffer.py @@ -375,7 +375,7 @@ def evaluate(self): # We connect each time just in case we block the wrappers with sqlite3.connect(self.sqlite_db) as conn: cur = conn.cursor() - resets = cur.executemany( + resets = cur.execute( """ SELECT reset, env_id FROM states