九头蛇在配置中组合列表

Question

假设我有以下水龙头配置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]。