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

remove jsonlines dependency

Browse source
This commit is contained in:
hjc-puro 2025-07-11 00:42:55 +00:00
parent 3e1eba6e92
commit 6c64df0226
1 changed files with 2 additions and 8 deletions
Download patch file Download diff file Expand all files Collapse all files

10
atroposlib/envs/base.py
View file

@ -670,12 +670,6 @@ class BaseEnv(ABC):
"data_dir_to_save_evals is not set, skipping evaluation logging"
)
return
import json
import os
import jsonlines
# Create directory if it doesn't exist
os.makedirs(self.config.data_dir_to_save_evals, exist_ok=True)
@ -868,9 +862,9 @@ class BaseEnv(ABC):
samples_filepath = os.path.join(
self.config.data_dir_to_save_evals, "samples.jsonl"
)
with jsonlines.open(samples_filepath, "w") as writer:
with open(samples_filepath, "w") as f:
for sample in samples:
writer.write(sample)
f.write(json.dumps(sample) + "\n")
print(f"Evaluation samples saved to {samples_filepath}")
@retry(