NousResearch/atropos
1
0
Fork 0
mirror of https://github.com/NousResearch/atropos.git synced 2026-04-19 12:57:58 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions

[pre-commit.ci] auto fixes from pre-commit.com hooks

Browse source 
for more information, see https://pre-commit.ci
This commit is contained in:
pre-commit-ci[bot] 2026-01-23 00:49:46 +00:00
parent 0a16fafadb
commit 655faa775c
27 changed files with 134 additions and 66 deletions
Download patch file Download diff file Expand all files Collapse all files

6
environments/eval_environments/vision_evals/docvqa_environment.py
View file

@ -5,10 +5,10 @@ import re
from typing import List, Tuple
from datasets import load_dataset
from environments.eval_environments.eval import EvalBase, eval_runner
from PIL import Image
from atroposlib.envs.server_handling.server_manager import ServerManager
from environments.eval_environments.eval import EvalBase, eval_runner
class DocVQA(EvalBase):
@ -142,7 +142,9 @@ Provide only the answer, as concisely as possible."""
return previous_row[-1]
async def run_item(self, server: ServerManager, data_item: dict) -> Tuple[dict, dict]:
async def run_item(
self, server: ServerManager, data_item: dict
) -> Tuple[dict, dict]:
try:
messages = self.build_messages(data_item)
completion = await self.chat_completion(server, messages)