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 duplicate code + add safety checks

Browse source
This commit is contained in:
Alireza 2026-02-09 10:50:29 +03:30
parent 7da681ec46
commit 6b92ee16ec
3 changed files with 4 additions and 6 deletions
Download patch file Download diff file Expand all files Collapse all files

3
atroposlib/utils/best_of_n_selection.py
View file

@ -36,9 +36,6 @@ def select_best_index(
raise ValueError("Primary and secondary score lists must have the same length.") raise ValueError("Primary and secondary score lists must have the same length.")
num_items = len(primary_scores) num_items = len(primary_scores)
if num_items == 0: # Should be caught by the first check, but as a safeguard.
raise ValueError("Input score lists cannot be empty.")
best_index = 0 best_index = 0
for i in range(1, num_items): for i in range(1, num_items):

5
environments/math_server.py
View file

@ -552,7 +552,10 @@ class MathEnv(BaseEnv):
i i
for i, score in enumerate(to_postprocess["scores"]) for i, score in enumerate(to_postprocess["scores"])
if score == 1.0 if score == 1.0
][0] ]
if len(pos_idx) == 0:
return None, to_backlog
pos_idx = pos_idx[0]
neg_idx = [ neg_idx = [
i i
for i, score in enumerate(to_postprocess["scores"]) for i, score in enumerate(to_postprocess["scores"])

2
environments/tool_use_interleaved_thinking.py
View file

@ -251,8 +251,6 @@ class InterleavedInlineEnv(BaseEnv):
if DEBUG: if DEBUG:
print(f"[DEBUG setup] kept {len(subset)} rows from Dataset") print(f"[DEBUG setup] kept {len(subset)} rows from Dataset")
split = full.train_test_split(test_size=0.02, seed=42)
split = full.train_test_split(test_size=0.02, seed=42) split = full.train_test_split(test_size=0.02, seed=42)
self.train, self.test = split["train"], split["test"] self.train, self.test = split["train"], split["test"]
self.train = self.train.shuffle(seed=int.from_bytes(os.urandom(2), "big")) self.train = self.train.shuffle(seed=int.from_bytes(os.urandom(2), "big"))