在类中使用 Hydra 配置

发布于 2025-01-18 01:34:48 字数 1296 浏览 3 评论 0原文

我正在尝试在项目中使用Hydra工具，并想将Decorator用于类功能，

import hydra
from hydra.core.config_store import ConfigStore

from src.config import RecordingConfig

cs = ConfigStore.instance()
cs.store(name="recording_config", node=RecordingConfig)

class HydraClassTest:
    @hydra.main(config_path="../src/conf/", config_name="conf")
    def __init__(self, conf: RecordingConfig):
        print(conf)

def main():
    HydraClassTest()

if __name__ == "__main__":
    main()

但是我遇到了这一错误

TypeError: __init__() missing 1 required positional argument: 'conf'

，我是否应该将配置从外部传递到课堂？（例如，通过在主函数上使用装饰器并将配置作为参数传递给初始化器，这起作用）
还是以错误的方式使用装饰器？如果是打算的，是否有一些设计原因为什么人们不想这样做？

我已经检查了是否正确地使用了配置，该配置是否正确地使用了装饰器。

import hydra
from hydra.core.config_store import ConfigStore

from src.config import RecordingConfig

cs = ConfigStore.instance()
cs.store(name="recording_config", node=RecordingConfig)

class HydraClassTest:
    def __init__(self, conf: RecordingConfig):
        print(conf)

@hydra.main(config_path="../src/conf/", config_name="conf")
def main(conf: RecordingConfig):
    HydraClassTest(conf)

if __name__ == "__main__":
    main()

这给了我预期的结果。

原文

I am trying to use the hydra tool in my project and would like to use the decorator for class functions

import hydra
from hydra.core.config_store import ConfigStore

from src.config import RecordingConfig

cs = ConfigStore.instance()
cs.store(name="recording_config", node=RecordingConfig)

class HydraClassTest:
    @hydra.main(config_path="../src/conf/", config_name="conf")
    def __init__(self, conf: RecordingConfig):
        print(conf)

def main():
    HydraClassTest()

if __name__ == "__main__":
    main()

But I get the error

TypeError: __init__() missing 1 required positional argument: 'conf'

Is this intended and should I pass the configuration from the outside to the class? (For example by using the decorator on the main function and passing the configuration as a parameter to the initializer, this works)
Or am using the decorator in a wrong way?
If it is intended, is there some design reason why one would not want to do it that way?

I have checked whether I used the decorator correctly by passing the configuration through the main function, that worked.

import hydra
from hydra.core.config_store import ConfigStore

from src.config import RecordingConfig

cs = ConfigStore.instance()
cs.store(name="recording_config", node=RecordingConfig)

class HydraClassTest:
    def __init__(self, conf: RecordingConfig):
        print(conf)

@hydra.main(config_path="../src/conf/", config_name="conf")
def main(conf: RecordingConfig):
    HydraClassTest(conf)

if __name__ == "__main__":
    main()

This gives me the expected result.

分享到QQ

分享到微博