mirror of https://github.com/PyRetri/PyRetri.git
32 lines
825 B
Python
32 lines
825 B
Python
|
# -*- coding: utf-8 -*-
|
||
|
|
|
||
|
|
from yacs.config import CfgNode
|
||
|
|
from copy import deepcopy
|
||
|
|
|
||
|
|
|
||
|
|
def _convert_dict_to_cfg(d: dict) -> CfgNode:
|
||
|
|
ret = CfgNode()
|
||
|
|
for key in d:
|
||
|
|
if isinstance(d[key], dict):
|
||
|
|
ret[key] = _convert_dict_to_cfg(d[key])
|
||
|
|
else:
|
||
|
|
ret[key] = d[key]
|
||
|
|
return ret
|
||
|
|
|
||
|
|
|
||
|
|
class SearchModules(dict):
|
||
|
|
r"""
|
||
|
|
This class defines the search args for one module, e.g., data process, feature extraction, feature
|
||
|
|
aggregation, feature process and query.
|
||
|
|
"""
|
||
|
|
def __init__(self):
|
||
|
|
super(SearchModules, self).__init__()
|
||
|
|
|
||
|
|
def add(self, name: str, value: dict):
|
||
|
|
self[name] = _convert_dict_to_cfg(value)
|
||
|
|
|
||
|
|
def check_valid(self, cfg: CfgNode):
|
||
|
|
cfg = deepcopy(cfg)
|
||
|
|
for module in self.values():
|
||
|
|
cfg.merge_from_other_cfg(module)
|