NousResearch/atropos
1
0
Fork 0
mirror of https://github.com/NousResearch/atropos.git synced 2026-04-23 16:54:56 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions

removing comments

Browse source
This commit is contained in:
Shannon Sands 2025-05-18 16:52:26 -07:00
parent 99a64f5bce
commit 034fbad7d5
1 changed files with 3 additions and 11 deletions
Download patch file Download diff file Expand all files Collapse all files

14
environments/tool_calling_server.py
View file

@ -64,13 +64,6 @@ class SingleToolCallingEnv(BaseEnv):
num_max_requests_at_once=32,
num_requests_for_eval=256,
),
# APIServerConfig(
# model_name="NousResearch/DeepHermes-3-Llama-3-8B-Preview",
# base_url="http://localhost:9005/v1",
# api_key="x",
# num_max_requests_at_once=32,
# num_requests_for_eval=256,
# ),
]
return env_config, server_configs
@ -301,7 +294,6 @@ class SingleToolCallingEnv(BaseEnv):
max_tokens=1024 * 15,
temperature=0.8, # Using temperature to get diverse responses
)
print(f"completions: {completions}")
to_score = list()
for i, completion_choice in enumerate(completions.choices):
@ -404,9 +396,9 @@ class SingleToolCallingEnv(BaseEnv):
# Apply linear penalty scaling from 1.0 down to 0.0
scores["scores"].append(1.0 - percentage_of_range)
# # Check if all scores are the same (no learning signal)
# if all(scores["scores"][0] == score for score in scores["scores"]):
# return None
# Check if all scores are the same (no learning signal)
if all(scores["scores"][0] == score for score in scores["scores"]):
return None
return scores