用pydantic验证类（Hydra-Core列表）

Question

1。上下文

如何验证pydantic的特定类别？

我正在使用 pydantic verate yaml列表参数由 hydra ，以后将其传递给建模例程。问题在于，Hydra字典不包含值列表，而是包含这些值的类。如何验证这些参数？

2。示例

在以下示例中，有2个文件：

• cfg.yaml包含要验证的参数
• main.py.py.py.py包含加载和验证cfg的指令.yaml

2.1配置文件cfg.yaml

params_list:
  - 10
  - 0
  - 20

2.2解析器/验证器文件main.py.py

import hydra
import pydantic
from omegaconf import DictConfig, OmegaConf
from typing import List

class Test(pydantic.BaseModel):
    params_list: List[int]

@hydra.main(config_path=".", config_name="cfg.yaml")
def go(cfg: DictConfig):
    parsed_cfg = Test(**cfg)
    print(parsed_cfg)

if __name__ == "__main__":
    go()

问题

3。执行时 python3 main.py.py出现以下错误

值不是有效列表（type = type_error.list）

这是因为Hydra具有用于处理列表的特定类，称为Omegaconf.listConfig.listConfig.listConfig，可以通过

print(type(cfg['params_list']))

在<<<<<代码> go（）函数定义。

4。指导

我知道我可能必须告诉pydantic来验证这一特定内容，但我只是不知道如何确切。

• 在这里 似乎有很多技巧我猜的任务。
• 另一个想法是为数据属性创建一个通用类型（例如params_list：generic），然后使用验证器装饰器将其转换为列表，符合行：

class ParamsList(pydantic.BaseModel):
  params_list: ???????? #i don't know that to do here
  @p.validator("params_list")
  @classmethod
    def validate_path(cls, v) -> None:
        """validate if it's a list"""
        if type(list(v)) != list:
            raise TypeError("It's not a list. Make it become a list")
        return list(v)

help！关于如何解决它的想法？

如何

1. 在第2.1和2.2节中描述的文件夹中重新创建示例。
2. 还可以创建一个带有软件包的insuert.txt文件pydantic和hydra-core
3. 在创建和激活Env之后，运行python3 main。 py

Answer 1

pydantic不接受dictconfig格式。当您尝试使用Pydantic模型解析Hydra配置时，您必须首先将DICTCONFIG转换为本机Python dict。您可以使用omegaconf.to_object（CFG）进行此操作。

我认为您使用Python 3.10或更高。注意version_base =“ 1.2”的使用以获取最新的Hydra版本。

这应该有效：

import hydra
import pydantic
from omegaconf import DictConfig, OmegaConf

class Test(pydantic.BaseModel):
    params_list: list[int]


@hydra.main(config_path=".", config_name="cfg.yaml", version_base="1.2")
def go(cfg: DictConfig):
    print(cfg)
    d_cfg = OmegaConf.to_object(cfg)
    parsed_cfg = Test(**d_cfg)
    print(parsed_cfg)


if __name__ == "__main__":
    go()