format

Author: mayinghan

eval_protocol/pytest/utils.py

@@ -257,9 +257,9 @@ async def retry_handler(failed_row: EvaluationRow):
             current_attempts = retry_counts.get(rollout_id, 0)
 
             if current_attempts >= max_retry:
-                assert (
-                    failed_row.rollout_status and failed_row.rollout_status.status == RolloutStatus.Status.ERROR
-                ), f"Rollout {failed_row.execution_metadata.rollout_id} did not fail with error status"
+                assert failed_row.rollout_status and failed_row.rollout_status.status == RolloutStatus.Status.ERROR, (
+                    f"Rollout {failed_row.execution_metadata.rollout_id} did not fail with error status"
+                )
                 failed_permanently.append(failed_row)
                 await queue.put(failed_row)  # put failed row on queue
                 return