python，Hydra，从YAML到列表的负载变量

Question

我得到了一个以下代码：

# settings.yaml

my_class:
  columns_id: ${oc.env:MY_LIST}
  all_columns: {"all_columns_as_str": ${oc.env:MY_LIST}}

我的变量为my_list = a，b，c

，然后我得到：

# hydra_conf.py

from dataclasses import dataclass

@dataclass
class MyClass:
  columns_id: list
  all_columns: dict[str, str]

然后我使用所有这些都像下面：

# main.py
import hydra
from hydra_conf import MyClass

@hydra.main(config_path="settings", config_name="settings")
def main(cfg: MyClass) -> None:
    assert isinstance(cfg.columns_id, list)
    assert isinstance(cfg.all_columns, dict)
    assert cfg.all_columns['all_columns_as_str'] == 'a,b,c'

我需要如何配置yml.file < /code>或我的hydra_conf.py load columns_id as list and all_columns 作为dict [str，str]？

Answer 1

尝试一下：

# settings.yaml
defaults:
  - my_class_schema
  - _self_

columns_id: ${oc.decode:"[${oc.env:MY_LIST}]"}
all_columns:
  all_columns_as_str: ${oc.env:MY_LIST}

# hydra_conf.py
from dataclasses import dataclass
from typing import Any
from hydra.core.config_store import ConfigStoreWithProvider

@dataclass
class MyClass:
    columns_id: list[Any]
    all_columns: dict[str, str]

cs = ConfigStoreWithProvider(__file__)
cs.store("my_class_schema", MyClass)

# main.py
import hydra
from omegaconf import DictConfig, OmegaConf
from hydra_conf import MyClass

@hydra.main(config_path=".", config_name="settings", version_base=None)
def main(cfg_: DictConfig) -> None:
    cfg: MyClass = OmegaConf.to_object(cfg_)
    assert isinstance(cfg, MyClass)
    assert isinstance(cfg.columns_id, list), f"{cfg.columns_id}"
    assert isinstance(cfg.all_columns, dict), f"{cfg.all_columns}"
    assert cfg.all_columns["all_columns_as_str"] == "a,b,c"

if __name__ == "__main__":
    main()

以下是这样做的方法：

• 在默认列表中包括my_class_schema，这意味着您的yaml文件将与myClass定义的架构合并。请参阅结构化configs “ https://hydra.cc/docs/tutorials/structured_config/schema/” rel =“ nofollow noreferrer”>结构化配置模式）有关此的更多信息。
• 字符串“ [$ {oc.env：my_list}]”在解决插值后，在中导致“ [a，b，c]”。
• oc.decode oc.decode resolver resolver解析字符串“ [a，b，c]”，导致[“ a”，“ b”，“ c”]。