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

use parse_http_rseponse

Browse source
This commit is contained in:
hjc-puro 2025-05-10 05:12:08 -04:00
parent a659217afe
commit e68df555ba
1 changed files with 2 additions and 2 deletions
Download patch file Download diff file Expand all files Collapse all files

4
atroposlib/envs/base.py
View file

@ -350,7 +350,7 @@ class BaseEnv(ABC):
async with session.get(
f"{self.config.rollout_server_url}/wandb_info"
) as resp:
data = await resp.json()
data = await parse_http_response(resp, logger)
self.wandb_group = data["group"]
self.wandb_project = data["project"]
if self.wandb_project is None:
@ -378,7 +378,7 @@ class BaseEnv(ABC):
"weight": self.config.inference_weight,
},
) as resp:
data = await resp.json()
data = await parse_http_response(resp, logger)
return data
except Exception as e:
logger.error(f"Error registering env: {e}")