NousResearch/atropos
1
0
Fork 0
mirror of https://github.com/NousResearch/atropos.git synced 2026-04-19 12:57:58 +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-15 11:42:42 +00:00
parent 53984580c8
commit 3d2d9e67fa
1 changed files with 4 additions and 6 deletions
Download patch file Download diff file Expand all files Collapse all files

10
atroposlib/envs/base.py
View file

@ -284,9 +284,7 @@ class BaseEnv(ABC):
"Handle env single method must be implemented in subclass "
)
async def collect_trajectories(
self, item: Item
) -> Tuple[
async def collect_trajectories(self, item: Item) -> Tuple[
Union[
Optional[ScoredDataGroup], List[Optional[ScoredDataGroup]], List[Any | None]
],
@ -1550,9 +1548,9 @@ class BaseEnv(ABC):
env_config_dict_base["ensure_scores_are_not_same"] = False
env_config_dict_base["include_messages"] = True
if env_config_dict_base.get("data_path_to_save_groups") is None:
env_config_dict_base[
"data_path_to_save_groups"
] = f"data/{cls.name or 'groups'}.jsonl"
env_config_dict_base["data_path_to_save_groups"] = (
f"data/{cls.name or 'groups'}.jsonl"
)
env_config_dict_base["use_wandb"] = True
env_config_dict = merge_dicts(