o qoi @sdZddlZddlZddlmZddlmZddlmZddlm Z ddl Z ddZ e j eje dd d Zd d ZddZe j ejee j ejee j e jededefddZdS)aKSerialization helper that writes YAML output. Note: This API is highly experimental, and primarily intended for dumping objects in a medium-easy-to-read format, using indentation/spaces instead of `printing.py`'s dot-separated paths. Please use `serialization.py` whenever you need to serialize Fiddle objects in a robust manner. N)Any)config)partial)taggingcCs ||S)N)represent_dictdumperdatar _/home/ubuntu/.local/lib/python3.10/site-packages/fiddle/_src/experimental/yaml_serialization.py_defaultdict_representer"s r fdl.ConfigcCsPt|j}d|vr tdtt|jt|jd|d<| d||S)z(Returns a YAML representation of `data`. __fn_or_cls__z]It is not supported to dump objects of functions/classes that have a __fn_or_cls__ parameter.)modulename!) dict __arguments__ ValueErrorinspect getmodule config_lib get_callable__name__ __qualname__represent_mapping)rr type_namevaluer r r _config_representer+s   rcCst||ddS)Nz fdl.Partial)r)rrr r r _partial_representer?srcCs |ddd|jD|jdS)Nz!fdl.TaggedValuecSsg|]}|jqSr )r).0tagr r r Gsz,_taggedvalue_representer..)tagsr)rr#rrr r r _taggedvalue_representerCs r$rreturncCs t|S)zReturns the YAML serialization of `value`. Args: value: The value to serialize. Raises: PyrefError: If an error is encountered while serializing a Python reference. )yaml safe_dump)rr r r dump_yamlTs r()r )__doc__ collectionsrtypingr fiddle._srcrrrrr&r SafeDumperadd_representer defaultdictrrr$ConfigPartialTaggedValueClsstrr(r r r r s*