o qoiX @sdZddlmZmZmZmZddlmZddlm Z edZ eede fee fZ dZ dej d ej fd d Z dd ej de deddfddZd ej fddZdS)z)Library for mutating Buildable instances.)CallableTypeTypeVarUnion)config) signaturesT.) __fn_or_cls__ __arguments____argument_tags____argument_history____signature_info__source destinationcCsPt|t|urtdt|dt|dtD] }t||t||qdS)aRChanges the internals of `destination` to be equivalent to `source`. Currently, this results in aliasing behavior, but this should not be relied upon. All the internals of `source` are aliased into `destination`. Args: source: The source configuration to pull internals from. destination: One of the buildables to swap. ztypes must match exactly: z vs .N)type TypeError_buildable_internals_keysobject __setattr__getattr)rrattrrP/home/ubuntu/.local/lib/python3.10/site-packages/fiddle/_src/mutate_buildable.pymove_buildable_internals%s  rF buildable new_callabledrop_invalid_argsreturnNcs|jD] }t|trtdqt|tjd}t |dt |d||j sWfdd|jD}|rW|rH|D]}t ||q?nt d|d|j d |d t |d ||jd |d S) a Updates ``config`` to build ``new_callable`` instead. When extending a base configuration, it can often be useful to swap one class for another. For example, an experiment may want to swap in a subclass that has augmented functionality. ``update_callable`` updates ``config`` in-place (preserving argument history). Args: buildable: A ``Buildable`` (e.g. a ``fdl.Config``) to mutate. new_callable: The new callable ``config`` should call when built. drop_invalid_args: If True, arguments that don't exist in the new callable will be removed from buildable. If False, raise an exception for such arguments. Raises: TypeError: if ``new_callable`` has varargs, or if there are arguments set on ``config`` that are invalid to pass to ``new_callable``. zKUpdate callable for configs with positional arguments is not supported yet.) signature __signature__r cs$g|]}|jvrt|tr|qSr) parameters isinstancestr).0arg new_signaturerr gs z#update_callable..zCannot switch to z (from z5) because the Buildable would have invalid arguments rr N)r keysr"intNotImplementedErrorr get_signature SignatureInforrhas_var_keyworddelattrrr r add_new_value)rrrkeynew_signature_info invalid_argsr%rr&rupdate_callable<s8    r4cKs"|D] \}}t|||qdS)arAssigns multiple arguments to ``buildable``. Although this function does not enable a caller to do something they can't already do with other syntax, this helper function can be useful when manipulating deeply nested configs. Example:: cfg = # ... fdl.assign(cfg.my.deeply.nested.child.object, arg_a=1, arg_b='b') The above code snippet is equivalent to:: cfg = # ... cfg.my.deeply.nested.child.object.arg_a = 1 cfg.my.deeply.nested.child.object.arg_b = 'b' Args: buildable: A ``Buildable`` (e.g. a ``fdl.Config``) to set values upon. **kwargs: The arguments and values to assign. N)itemssetattr)rkwargsnamevaluerrrassignzsr:)F)__doc__typingrrrr fiddle._srcr config_librrTypeOrCallableProducingTr Buildablerboolr4r:rrrrs.     >