o -wiM @sddlZddlZddlZddlmZmZmZddlZddlZddl m Z ddl m Z ddl mZedZGdd d e Zd d Zd d ZddZddZ ddededeeeeffddZdededefddZdS)N)AnyDictOptional)Error) load_yaml) filesystemwandbc@s eZdZdS) ConfigErrorN)__name__ __module__ __qualname__rrV/home/ubuntu/sommelier/.venv/lib/python3.10/site-packages/wandb/sdk/lib/config_util.pyr sr cCs.t}|D]}tdt|jd||j<q|S)N)descvalue)dictjsonloads value_jsonkey)obj_listditemrrrdict_from_proto_listsrcCs(t}|D] \}}|d||<q|SNr)ritems) config_dictrkvrrrdict_strip_value_dictsr cCsBt}|D]}t|j}t|trd|vrq|d||j<q|Sr)rrrr isinstancer)rrr possible_dictrrrdict_no_value_from_proto_list#s r#c Cs~d}|r|dtj|tjddddd7}|d}ttj|t |d}| |WddS1s8wYdS)Nswandb_version: 1s FTzutf-8)Dumperdefault_flow_style allow_unicodeencoding sort_keysw) yamldump SafeDumperdecodermkdir_exists_okospathdirnameopenwrite)config_filenamersdata conf_filerrrsave_config_file_from_dict/s   "r8Ffilename must_existreturncCstj|s|rtd|dtd|dSzt|}Wnty.td|wzt|}Wn t j j yBtdw|durNt ddS|dd}|dur`|dkr`td t}|D] \}}|d ||<qg|S) Nz config file z doesn't existz no default config file found in zCouldn't read config file: zInvalid YAML in config yamlzWFound an empty default config file (config-defaults.yaml). Proceeding with no defaults. wandb_versionrzUnknown config versionr)r/r0existsr loggerdebugr2OSErrorrr*parser ParserErrorr termwarnpoprr)r9r:r7loadedconfig_versionr6rrrrrdict_from_config_file@s6     rGdestsrccCsN|D] \}}t|tr ||vr t||tr t|||q|||<q|S)zBRecursively merge two dictionaries. Similar to Lodash's _.merge().)rr!r merge_dicts)rHrIrrrrrrJ^s   rJ)F)rloggingr/typingrrrr*r wandb.errorsr wandb.utilrr getLoggerr>r rr r#r8strboolrGrrJrrrrs0