Hydra 组合配置中的列表

Question

假设我有以下 Hydra 配置 test.yaml：

list1 : [0]
list2 : [1,2,3]

是否可以将 list1 和 list2 合并到包含 [ 的新列表中0,1,2,3]，也许使用变量插值？

这是九头蛇代码：

import hydra
from omegaconf import OmegaConf
@hydra.main(config_name="test.yaml", config_path="./")
def main(cfg):
    OmegaConf.resolve(cfg)
    print(cfg)
if __name__ == "__main__":
    main()

尝试（失败）：

list1 : [0]
list2 : [1,2,3]
list3 : 
  - ${list1} 
  - ${list2}

list3给出[0,[1,2,3]]

我想要这个的原因是因为我有一些列表其他配置文件中的长度未知，并且想要合并它们以创建对象创建的参数。我不想在代码中执行此操作，而是直接依赖 Hydras 对象实例化（否则我将在各处进行列表合并！）。

Answer 1

事实证明这并不太困难，我们确实可以依靠 OmegaConfs 变量插值与自定义解析器。

import hydra
from omegaconf import OmegaConf
# custom list merge resolver
OmegaConf.register_new_resolver("merge", lambda x, y : x + y)

@hydra.main(config_name="test.yaml", config_path="./")
def main(cfg):
    OmegaConf.resolve(cfg)
    print(cfg)

if __name__ == "__main__":
    main()

配置文件 test.yaml

list1 : [0]
list2 : [1,2,3]

list3 : ${merge:${list1},${list2}}

list3 现在是所需的 [0,1,2,3]。