check_marl_grouping¶
- torchrl.envs.check_marl_grouping(group_map: dict[str, list[str]], agent_names: list[str])[source]¶
檢查 MARL 組對映。
對 MARL 環境的組對映執行檢查以評估其有效性。如果 group_map 無效，則引發錯誤。
- 引數:
group_map (Dict[str, List[str]]) – 將組名稱對映到組中智慧體名稱列表的組對映
agent_names (List[str]) – 環境中所有智慧體名稱的列表
示例
>>> from torchrl.envs.utils import MarlGroupMapType, check_marl_grouping >>> agent_names = ["agent_0", "agent_1", "agent_2"] >>> check_marl_grouping(MarlGroupMapType.ALL_IN_ONE_GROUP.get_group_map(agent_names), agent_names)
