Merge pull request #505 from tureba/fix-error-reporting-logs-of-runner

fix error reporting of runner
erikbern · May 6, 2024 · 1c3f702 · 1c3f702
2 parents 75043ab + ceb8a3e
commit 1c3f702
Showing 1 changed file with 2 additions and 1 deletion.
diff --git a/ann_benchmarks/runner.py b/ann_benchmarks/runner.py
@@ -381,7 +381,8 @@ def _handle_container_return_value(
         msg = msg.format(exit_code)
 
     if exit_code not in [0, None]:
-        logger.error(colors.color(container.logs().decode(), fg="red"))
+        for line in container.logs(stream=True):
+            logger.error(colors.color(line.decode(), fg="red"))
         logger.error(msg)
     else:
         logger.info(msg)