Add Reflexion prompts for FrozenLake + tweak ToT to allow for recordings

JetBrains-Research · Mar 18, 2024 · 8fe84ae · 8fe84ae
1 parent e798a65
commit 8fe84ae
Show file tree

Hide file tree

Showing 10 changed files with 1,590 additions and 146 deletions.
diff --git a/environments/frozen_lake/common/tools.py b/environments/frozen_lake/common/tools.py
@@ -43,7 +43,7 @@ def _convert_frozenlake_observation_to_position(
     ) -> Tuple[int, int]:
         # FrozenLake: observation = current_row * nrow + current_col
         current_row, current_col = observation // nrow, observation % nrow
-        return (current_row, current_col)
+        return current_col, current_row
 
     @staticmethod
     def _convert_direction_to_frozenlake(direction: str) -> int: