Update action_execution_server.py

SmartManoj · Nov 6, 2024 · 6eab6e3 · 6eab6e3
1 parent ef83707
commit 6eab6e3
Showing 1 changed file with 5 additions and 1 deletion.
diff --git a/openhands/runtime/action_execution_server.py b/openhands/runtime/action_execution_server.py
@@ -258,7 +258,11 @@ async def run_ipython(self, action: IPythonRunCellAction) -> Observation:
                 obs: Observation = ErrorObservation(
                     "[The content in this file is absolutely correct. Also, you can't modify this test file. You must pass this test case. You should correct the codebase instead.]"
                 )
-            elif action.code == self.last_code and self.is_last_code_error:
+            elif (
+                (os.getenv('SWE_BENCH') == '1' or 1)
+                and action.code == self.last_code
+                and self.is_last_code_error
+            ):
                 obs = ErrorObservation(
                     '[You are trying to run the same code twice. Please focus and run the correct code.]'
                 )