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

fix: pass num_steps to register_to_api

Browse source 
Signed-off-by: Ocheretovich Oksana <ocheretovich@gmail.com>
This commit is contained in:
Ocheretovich Oksana 2026-02-24 11:22:18 +02:00
parent 329a233bba
commit aec5552db6
1 changed files with 1 additions and 1 deletions
Download patch file Download diff file Expand all files Collapse all files

2
atroposlib/cli/dpo.py
View file

@ -223,7 +223,7 @@ async def dpo_data_grabber(
count += 1
return count
await register_to_api(group_size, max_token_len, api_url)
await register_to_api(group_size, max_token_len, api_url, num_seqs_to_save)
if os.path.exists(filepath) and not append_to_previous:
raise ValueError("File already exists and append_to_previous is False.")
with open(filepath, "w" if not append_to_previous else "a") as f: