Updating python block

utils.py

@@ -45,7 +45,8 @@ def _execute_python(code: str, stdin: str) -> Tuple[str, str, str]:
     with capture_output() as (stdout, stderr):
         try:
             inputs = iter(stdin.splitlines())
-            local_vars = {"input": lambda _: next(inputs, "")}
+            input_override = lambda prompt='': next(inputs, '')
+            local_vars = {"input": input_override}
             exec(code, {}, local_vars)
             return stdout.getvalue().strip(), stderr.getvalue().strip(), None
         except Exception as e: