move from for-loop to iter

modal-labs · Sep 17, 2024 · 79471cc · 79471cc
1 parent 7088f66
commit 79471cc
Showing 1 changed file with 3 additions and 3 deletions.
diff --git a/08_advanced/code_interpreter.py b/08_advanced/code_interpreter.py
@@ -80,9 +80,9 @@ def _exec_cell_remote(self, code: str) -> ExecutionResult:
         self.sb.stdin.write(code.encode("utf-8"))
         self.sb.stdin.write(b"\n")
         self.sb.stdin.drain()
-        for message in self.sb.stdout:
-            if message.strip() == NULL_MARKER:
-                return ExecutionResult(None)
+        message = next(iter(self.sb.stdout))
+        if message.strip() == NULL_MARKER:
+            return ExecutionResult(None)
         return ExecutionResult(message)
 
     def _exec_cell_local(self, code: str) -> ExecutionResult: