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

[pre-commit.ci] auto fixes from pre-commit.com hooks

Browse source 
for more information, see https://pre-commit.ci
This commit is contained in:
pre-commit-ci[bot] 2025-07-04 14:34:33 +00:00
parent 14c70c0e68
commit ee5257522a
1 changed files with 6 additions and 2 deletions
Download patch file Download diff file Expand all files Collapse all files

8
atroposlib/envs/base.py
View file

@ -394,7 +394,9 @@ class BaseEnv(ABC):
# Setup wandb getting the group and project via the server
while self.wandb_project is None:
async with aiohttp.ClientSession() as session:
async with session.get(f"{self.config.rollout_server_url}/wandb_info") as resp:
async with session.get(
f"{self.config.rollout_server_url}/wandb_info"
) as resp:
data = await parse_http_response(resp, logger)
self.wandb_group = data["group"]
self.wandb_project = data["project"]
@ -407,7 +409,9 @@ class BaseEnv(ABC):
if self.config.wandb_name:
random_id = "".join(random.choices(string.ascii_lowercase, k=6))
current_date = datetime.now().strftime("%Y-%m-%d")
wandb_run_name = f"{self.config.wandb_name}-{current_date}-{random_id}"
wandb_run_name = (
f"{self.config.wandb_name}-{current_date}-{random_id}"
)
wandb.init(
name=wandb_run_name,