o qoi@sdZddlZddlmZmZejGdddZeddded dded dd d d d edddedddedddedddedddedddedddeddded ddedddd Zdeded dfd!d"Z d#ed$ed efd%d&Z dS)'z>Utilities for special overrides for codegen and serialization.N)DictOptionalc@sFeZdZUdZeed<dZeeed<ej e dZ e eefed<dS)SpecialOverridesz(Overrides for codegen and serialization. module_nameNmodule_import_alias)default_factory#migrated_symbol_destination_modules) __name__ __module__ __qualname____doc__str__annotations__rr dataclassesfielddictrrrrQ/home/ubuntu/.local/lib/python3.10/site-packages/fiddle/_src/special_overrides.pyrs r fiddle.configzimport fiddle as fdl)rrfiddlefiddle._src.configfiddle._src.partial)Partial ArgFactory)rrrfiddle._src.mutate_buildablezfiddle._src.copyingfiddle._src.experimentalzfrom fiddle import experimental$fiddle._src.experimental.auto_configz+from fiddle.experimental import auto_configfiddle._src.arg_factoryzfrom fiddle import arg_factoryfiddle._src.codegenzfrom fiddle import codegen+fiddle._src.codegen.py_val_to_cst_converterz1from fiddle.codgen import py_val_to_cst_converterfiddle._src.daglishzfrom fiddle import daglishfiddle._src.tagging) rrrrzfiddle._src.copyrrrrrr rr!rspecial_overridesreturncCs |t|<dS)N)SPECIAL_OVERRIDES_MAP)rr"rrrregister_special_override`s r%modulesymbolcCs(|tvr|t|jvrt|j|S|S)a.Optionally get overridden module qual name for migrated symbol. Args: module: the module where the symbol may have been originally present. symbol: the symbol that may have been migrated Returns: Overridden module name if the symbol was migrated, else the supplied module name. )r$r)r&r'rrr;maybe_get_module_override_for_migrated_serialization_symbolfsr() r rtypingrr dataclassrr$r r%r(rrrrs;