B
|
ú{`H ã @ sf d Z ddlZddlZddlmZ ddlZddlmZ ddl m
|
Z
|
dZ G dd deZed d
|
d Z dS ) z9
|
@author: l1aoxingyu
|
@contact: sherlockliao01@gmail.com
|
é N)ÚAny)ÚCfgNode)Ú PathManagerZ_BASE_c sn e Zd ZdZedeedddZdeedddZ fd d
|
Z e
|
d fd d Z ee d fdd Z ZS )r aM
|
Our own extended version of :class:`yacs.config.CfgNode`.
|
It contains the following extra features:
|
1. The :meth:`merge_from_file` method supports the "_BASE_" key,
|
which allows the new CfgNode to inherit all the attributes from the
|
base configuration file.
|
2. Keys that start with "COMPUTED_" are treated as insertion-only
|
"computed" attributes. They can be inserted regardless of whether
|
the CfgNode is frozen or not.
|
3. With "allow_unsafe=True", it supports pyyaml tags that evaluate
|
expressions in config. See examples in
|
https://pyyaml.org/wiki/PyYAMLDocumentation#yaml-tags-and-python-types
|
Note that this may lead to arbitrary code execution: you must not
|
load a config file from untrusted sources before manually inspecting
|
the content of the file.
|
F)ÚfilenameÚ allow_unsafec s t | d¡v}yt |¡}W n` tjjk
|
r| |s6 t t¡}| d
|
| ¡¡ | ¡ t| d}t |¡}W dQ R X Y nX W dQ R X fdd t |kr
|
|t }| d¡r¼tj |¡}tt|jddd gsètj tj | ¡|¡}tj||d
|
}|t = || |S |S ) a
|
Just like `yaml.load(open(filename))`, but inherit attributes from its
|
`_BASE_`.
|
Args:
|
filename (str): the file name of the current config. Will be used to
|
find the base config file.
|
allow_unsafe (bool): whether to allow loading the config file with
|
`yaml.unsafe_load`.
|
Returns:
|
(dict): the loaded yaml
|
ÚrzlLoading config {} with yaml.unsafe_load. Your machine may be at risk if the file contains malicious content.Nc s` xZ| ¡ D ]N\}}t|trP||krPt|| ts@td |¡ ||| q
|
|||< q
|
W d S )Nz"Cannot inherit key '{}' from base!)ÚitemsÚ
|
isinstanceÚdictÚAssertionErrorÚformat)ÚaÚbÚkÚv)Úmerge_a_into_b© ú./config/config.pyr C s z3CfgNode.load_yaml_with_base.<locals>.merge_a_into_bú~ú/zhttps://zhttp://)r )r ÚopenÚyamlZ safe_loadÚ constructorZConstructorErrorÚloggingÚ getLoggerÚ__name__Úwarningr ÚcloseZ unsafe_loadÚBASE_KEYÚ
|
startswithÚosÚpathÚ
|
expanduserÚanyÚmapÚjoinÚdirnamer Úload_yaml_with_base)r r ÚfÚcfgÚloggerZ base_cfg_fileZbase_cfgr )r r r' % s8
|
$
|
|
|
|
zCfgNode.load_yaml_with_base)Ú cfg_filenamer c C s t j||d}t| |}dS )zí
|
Merge configs from a given yaml file.
|
Args:
|
cfg_filename: the file name of the yaml config.
|
allow_unsafe: whether to allow loading the config file with
|
`yaml.unsafe_load`.
|
)r N)r r' Útype)Úselfr+ r Z
|
loaded_cfgr r r Úmerge_from_fileb s
|
zCfgNode.merge_from_filec s" t |kstd t ¡t |¡S )zO
|
Args:
|
cfg_other (CfgNode): configs to merge from.
|
z0The reserved key '{}' can only be used in files!)r r r ÚsuperÚmerge_from_other_cfg)r- Z cfg_other)Ú __class__r r r0 q s
|
z CfgNode.merge_from_other_cfg)Úcfg_listc s4 t |ddd
}t|ks(td t¡t |¡S )zS
|
Args:
|
cfg_list (list): list of configs to merge from.
|
r Né z0The reserved key '{}' can only be used in files!)Úsetr r r r/ Úmerge_from_list)r- r2 Úkeys)r1 r r r5 } s
|
zCfgNode.merge_from_list)ÚnameÚvalc sT | d¡rB|| kr8| | }||kr&d S td |||¡|| |< nt ||¡ d S )NZ COMPUTED_zOComputed attributed '{}' already exists with a different value! old={}, new={}.)r ÚKeyErrorr r/ Ú __setattr__)r- r7 r8 Zold_val)r1 r r r: s
|
|
zCfgNode.__setattr__)F)F)r Ú
|
__module__Ú __qualname__Ú__doc__Ú staticmethodÚstrÚboolr' r. r0 Úlistr5 r r: Ú __classcell__r r )r1 r r s < r )Úreturnc C s ddl m} | ¡ S )z]
|
Get a copy of the default config.
|
Returns:
|
a fastreid CfgNode instance.
|
é )Ú_C)ÚdefaultsrE Úclone)rE r r r Úget_cfg s rH ) r= r r Útypingr r Z yacs.configr Z_CfgNodeÚ utils.file_ior r rH r r r r Ú<module> s
|
