Function handle_process_completion

run_evals.py:146–179

Handles the completion of a process, checks for errors, cleans up stderr_file, and logs messages.

(process, stderr_file, config_path: Path, verbose: bool, gpu_id: Optional[int] = None)

def handle_process_completion(process, stderr_file, config_path: Path, verbose: bool, gpu_id: Optional[int] = None):
    """Handles the completion of a process, checks for errors, cleans up stderr_file, and logs messages."""
    returncode = process.returncode

    # Read and clean up stderr output
    if stderr_file is not None:
        stderr_file.seek(0)
        error_output = stderr_file.read()
        stderr_file.close()
        os.unlink(stderr_file.name)  # Delete the temp file
    else:
        error_output = "Error output was displayed above."

    # Construct job identifier
    if gpu_id is not None:
        job_identifier = f"Job on GPU {gpu_id} for {config_path.name}"
    else:
        job_identifier = f"Job for {config_path.name}"

    if returncode != 0:
        # The process exited with an error
        if verbose:
            print(f"{job_identifier} failed with return code {returncode}")
            print("Error Output:")
            print(error_output)
        else:
            console.print(f"[red]{job_identifier} failed with return code {returncode}[/red]")
            console.print(f"[red]Error Output:[/red]\n{error_output}")
    else:
        # The process completed successfully
        if verbose:
            print(f"{job_identifier} has finished successfully.")
        else:
            console.log(f"{job_identifier} has finished successfully.")


def run_job(