NousResearch/atropos
1
0
Fork 0
mirror of https://github.com/NousResearch/atropos.git synced 2026-04-26 17:13:09 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions

remove use_api in env_manager, log config to wandb

Browse source
This commit is contained in:
hjc-puro 2025-05-02 08:52:28 -07:00
parent 8c9752b731
commit 6661e286c4
1 changed files with 7 additions and 7 deletions
Download patch file Download diff file Expand all files Collapse all files

14
atroposlib/envs/base.py
View file

@ -781,17 +781,12 @@ class BaseEnv(ABC):
)
)
async def env_manager(self, use_api=True):
async def env_manager(self):
"""
Rollout manager
"""
await self.setup()
await self.setup_wandb()
if not use_api:
await self.add_train_workers()
return
await self.register_env()
await self.get_server_info()
# Wait for other instances to get setup :)
@ -846,7 +841,12 @@ class BaseEnv(ABC):
random_id = "".join(random.choices(string.ascii_lowercase, k=6))
current_date = datetime.now().strftime("%Y-%m-%d")
wandb_run_name = f"{self.name}-{current_date}-{random_id}"
wandb.init(project=self.wandb_project, name=wandb_run_name)
wandb.init(
project=self.wandb_project,
name=wandb_run_name,
group=self.wandb_group,
config=self.config.model_dump(),
)
# Initialize the processing
self.curr_step = 0