NousResearch/atropos
1
0
Fork 0
mirror of https://github.com/NousResearch/atropos.git synced 2026-04-25 17:10:42 +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-04-06 20:55:42 +00:00
parent 48dcd64299
commit 51b4aa4858
6 changed files with 147 additions and 99 deletions
Download patch file Download diff file Expand all files Collapse all files

10
atroposlib/api/server.py
View file

@ -213,13 +213,13 @@ def _process_scored_data(scored_data: ScoredData) -> Dict[str, Any]:
if actual_group_size != expected_group_size:
buffer = app.state.buffer.setdefault(env_id, [])
buffer.append(data_dict)
if hasattr(app.state, "shm_buffer") and app.state.shm_buffer:
for i in range(len(scored_data.tokens)):
app.state.shm_buffer.write_trajectory(
tokens=scored_data.tokens[i],
score=scored_data.scores[i],
metadata={"env_id": env_id}
metadata={"env_id": env_id},
)
app.state.queue.append(data_dict)
@ -266,7 +266,7 @@ async def register(registration: Registration):
app.state.requesters = []
app.state.requesters.append(uuid.uuid4().int)
# Pin-hole SHM initialization
shm_name = f"atropos_shm_{app.state.group}"
try:
@ -274,7 +274,7 @@ async def register(registration: Registration):
name=shm_name,
size=app.state.batchsize * 10,
entry_size=app.state.max_token_len,
create=True
create=True,
)
except Exception as e:
logger.error(f"SHM Buffer Init Failed: {e}")
@ -282,7 +282,7 @@ async def register(registration: Registration):
return {
"uuid": app.state.requesters[-1],
"shm_handle": shm_name if app.state.shm_buffer else None
"shm_handle": shm_name if app.state.shm_buffer else None,
}