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

gsm8k eval example

Browse source
This commit is contained in:
hjc-puro 2025-07-11 00:22:36 +00:00
parent afe4fc318f
commit e601251893
2 changed files with 6 additions and 1 deletions
Download patch file Download diff file Expand all files Collapse all files

6
atroposlib/envs/base.py
View file

@ -688,7 +688,13 @@ class BaseEnv(ABC):
else:
task_name = f"{self.__class__.__name__}_eval"
if model_name is None:
# Try to get model name from config first, then from server configs
model_name = getattr(self.config, 'model_name', None)
if model_name is None and hasattr(self, 'server') and self.server.servers:
# Get model name from first server config
first_server = self.server.servers[0]
if hasattr(first_server, 'config') and hasattr(first_server.config, 'model_name'):
model_name = first_server.config.model_name
if start_time is None:
start_time = time.time()
if end_time is None: