2Vh0UdZddlZddlZddlZddlZddlZddlmZddlm Z da dZ dZ ejZejaejadZd Zd ZGd d ZGd dZGddZGddeZGddZ d!dZej6dZeddgdZ d"dZeddg d"dZdZ dZ!d Z"y)#z,Legacy serialization logic for Keras models.N) keras_export)object_registrationFzlayer was saved without configshared_object_idc$ttddS)zFGet whether shared object handling is disabled in a threadsafe manner.disabledF)getattrSHARED_OBJECT_DISABLEDU/home/dcms/DCMS/lib/python3.12/site-packages/keras/src/legacy/saving/serialization.py_shared_object_disabledr %s ):u ==r c4ttdtS)BGet the current shared object saving scope in a threadsafe manner.scope)rSHARED_OBJECT_LOADINGNoopLoadingScoper r r _shared_object_loading_scoper*s ('3C3E FFr c$ttddS)rrN)rSHARED_OBJECT_SAVINGr r r _shared_object_saving_scoper/s '$ 77r ceZdZdZdZdZy)DisableSharedObjectScopezA context manager for disabling handling of shared objects. Disables shared object handling for both saving and loading. Created primarily for use with `clone_model`, which does extra surgery that is incompatible with shared objects. cVdt_t|_t |_yNT)r rr_orig_loading_scoper_orig_saving_scopeselfs r __enter__z"DisableSharedObjectScope.__enter__=s!*.'#?#A "="?r cndt_|jt_|j t _y)NF)r rrrrrrrargskwargss r __exit__z!DisableSharedObjectScope.__exit__Bs)*/'&*&>&>#%)%<%<"r N)__name__ __module__ __qualname____doc__rr$r r r rr4s@ =r rceZdZdZdZdZy)rzThe default shared object loading scope. It does nothing. Created to simplify serialization code that doesn't care about shared objects (e.g. when serializing a single object). cyNr )runused_object_ids r getzNoopLoadingScope.getOsr cyr+r r object_idobjs r setzNoopLoadingScope.setRs r N)r%r&r'r(r-r2r r r rrHs  r rc(eZdZdZdZdZdZdZy)SharedObjectLoadingScopeaWA context manager for keeping track of loaded objects. During the deserialization process, we may come across objects that are shared across multiple layers. In order to accurately restore the network structure to its original state, `SharedObjectLoadingScope` allows us to re-use shared objects rather than cloning them. cRtr tS|t_i|_|Sr+)r rrr_obj_ids_to_objrs r rz"SharedObjectLoadingScope.__enter___s) " $#% %'+#! r c>|y|jj|S)aGiven a shared object ID, returns a previously instantiated object. Args: object_id: shared object ID to use when attempting to find already-loaded object. Returns: The object, if we've seen this ID before. Else, `None`. N)r6r-)rr0s r r-zSharedObjectLoadingScope.geths$   ##'' 22r c(|y||j|<y)z>Q &*nnD" # !r )r%r&r'r(rArF __classcell__)rCs@r r;r;s 0 r r;c(eZdZdZdZdZdZdZy)SharedObjectSavingScopez6Keeps track of shared object configs when serializing.ctrytd|_tSd|_|t_t j |_d|_|S)NTFr) r r _passthroughrrweakrefWeakKeyDictionary_shared_objects_config_next_idrs r rz!SharedObjectSavingScope.__enter__sU " $ ' ( 4 $D .0 0 %D %)"&-&?&?&A#  r cp |j|}|j|S#ttf$rYywxYw)aGets a `SharedObjectConfig` if one has already been seen for `obj`. Args: obj: The object for which to retrieve the `SharedObjectConfig`. Returns: The SharedObjectConfig for a given object, if already seen. Else, `None`. N)rN TypeErrorKeyErrorrF)rr1shared_object_configs r get_configz"SharedObjectSavingScope.get_configsG #'#>#>s#C  002##8$    s #55ct||j}|xjdz c_ ||j|<|S#t$rY|SwxYw)z3Create a new SharedObjectConfig for a given object.r>)r;rOrNrQ)rrBr1rSs r create_configz%SharedObjectSavingScope.create_configsZ1+t}}M   /CD ' ' ,$#   ##  s> A  A c6t|dds dt_yy)NrKF)rrrr!s r r$z SharedObjectSavingScope.__exit__st^U3)- &4r N)r%r&r'r(rrTrVr$r r r rIrIs@,$* $.r rIc||d}| ||t<t9|7tj|}|tj||S|S|S)z=Returns the serialization of the class with the given config.) class_nameconfig)rErrTrV)cls_name cls_configr1rrBrSs r serialize_keras_class_and_configr]sl"*Z@K #)9 %& #$0S_:<GGL  '.0>>{CP P## r c#8Kt} dad|ay#|awxYwwr)_SKIP_FAILED_SERIALIZATION)prevs r skip_failed_serializationras' &D*%)" %)"T"sz*keras.legacy.saving.serialize_keras_objectz)keras.utils.legacy.serialize_keras_objectctj|}|yt|drtj|j } |j }i}|jD]Q\}}t|tr|||< t|}t|trt|tsd|d<|||<Stj|j }t|||St|drtj|St!d|d#t$r$}trt|tdicYd}~S|d}~wwxYw#t $r|||<YwxYw)aSerialize a Keras object into a JSON-compatible representation. Calls to `serialize_keras_object` while underneath the `SharedObjectSavingScope` context manager will cause any objects re-used across multiple layers to be saved with a special shared object ID. This allows the network to be re-created properly during deserialization. Args: instance: The object to serialize. Returns: A dict-like, JSON-compatible representation of the object's config. NrTT__passive_serialization__r%zCannot serialize z- because it doesn't implement `get_config()`.)inspectunwraphasattrrget_registered_namerCrTNotImplementedErrorr_r]_LAYER_UNDEFINED_CONFIG_KEYitems isinstancestrserialize_keras_objectdict ValueError)instancenamerZeserialization_configkeyitemserialized_items r rmrm s},~~h'Hx&"66x7I7IJ ((*F " 1IC$$,0$S) 1"8">ot4Z$>DHO$?@,;$S) 1"#66x7I7IJ/ &  x$"66x@@  H:&  =# )76=G  * 1,0$S) 1s6D5E E D?7E=D??EEEcjt|trd|vsd|vrtd|d|d}tj|||}|td|d|d|d}t|t r||fSi}|j D]\}} |dk(r| ||<t| trd | vrt| ||d ||<8t| tsItjtj| |sstj| |||<|j D] \}} ||||<||fS) z@Returns the class name and config for a serialized keras object.rYrZzImproper config format for zB. Expecting python dict contains `class_name` and `config` as keysUnknown : ''. Please ensure you are using a `keras.utils.custom_object_scope` and that this object is included in the scope. See https://www.tensorflow.org/guide/keras/save_and_serialize#registering_the_custom_object for details.rqrc config_item)module_objectscustom_objectsprintable_module_name) rkrnrorget_registered_objectlistrjdeserialize_keras_objectrlrd isfunction) rZr|r}r~rYclsr\deserialized_objectsrtrus r ,class_and_config_for_serialized_keras_objectrKs vt $ v % 6 !)&2O O   %J  3 3NN C {,-S =: :  !J*d#Z  %%' T &= )-  % d #(Ct(K(@--&3 )  %c "w'9'9  5 5dN K( $99$O ! %7<*//14 T.s3 34  r z,keras.legacy.saving.deserialize_keras_objectz+keras.utils.legacy.deserialize_keras_objectc|yt|tr/|}t||||\}}|jt}t j|}||St |drtj|j} |xsi}t|dd}d| jvr&|j|itj|} n\tj|5|j|} dddn,|xsi}tj|5|d i|} dddt j| | St|t r|} |r| |vr|j| } n| tj"j$vrtj"j$| } nK| tj&vrtj&| } n%|j| } | t)d|d| dtj*| r| S| Stj,|r|St)d |d |#1swYxYw#1swY(xYw) aTurns the serialized form of a Keras object back into an actual object. This function is for mid-level library implementers rather than end users. Importantly, this utility requires you to provide the dict of `module_objects` to use for looking up the object config; this is not populated by default. If you need a deserialization utility that has preexisting knowledge of built-in Keras objects, use e.g. `keras.layers.deserialize(config)`, `keras.metrics.deserialize(config)`, etc. Calling `deserialize_keras_object` while underneath the `SharedObjectLoadingScope` context manager will cause any already-seen shared objects to be returned as-is rather than creating a new object. Args: identifier: the serialized form of the object. module_objects: A dictionary of built-in objects to look the name up in. Generally, `module_objects` is provided by midlevel library implementers. custom_objects: A dictionary of custom objects to look the name up in. Generally, `custom_objects` is provided by the end user. printable_module_name: A human-readable string representing the type of the object. Printed in case of exception. Returns: The deserialized object. Example: A mid-level library implementer might want to implement a utility for retrieving an object from its config, as such: ```python def deserialize(config, custom_objects=None): return deserialize_keras_object( identifier, module_objects=globals(), custom_objects=custom_objects, name="MyObjectType", ) ``` This is how e.g. `keras.layers.deserialize()` is implemented. N from_configzkeras.r})r}rxryrzzCould not interpret serialized z: r )rkrnrr-rErrfrdgetfullargspecr_find_replace_nested_dictr"rGLOBAL_CUSTOM_OBJECTSCustomObjectScoper2rl_THREAD_LOCAL_CUSTOM_OBJECTS__dict___GLOBAL_CUSTOM_OBJECTSroisclassr) identifierr|r}r~rZrr\r shared_objectarg_specdeserialized_obj object_namer1s r rrst*d#H NN4I j "::&7846::;KL  $ 3 &--coo>H+1rN3HhJ 8==0#&??$-CC$($$3$ )::>JC'*z'B$CC ,1rN$66~F 5##4  5 %&**+;=MN J $ k^; $$[1C "??HH I&BBKKC/FF F%<<[IC $$[1C{ 45S FBB ??3 5L   J ' -$%R | 5  cCC 5 5s!H9 I9IIc6t|txrt|vS)z=Determines whether config appears to be a valid layer config.)rkrnri)rZs r validate_configr.s 64 N%@%Nr ct|ddS)z:Check if a method is decorated with the `default` wrapper. _is_defaultF)r)methods r is_defaultr5s 6=% 00r c~tj|}|j||}tj|}|Sr+)jsondumpsreplaceloads)rZfindrdict_strs r rr:s5zz&!Hg.H ZZ !F Mr )NN)NNobject)#r( contextlibrdr threadingrLkeras.src.api_exportrkeras.src.savingrr_rirElocalr rrr rrrrr4rnr;rIr]contextmanagerrarmrrrrrr r r rsG2 .0 #?'(*' )&y( > G 8 ==(   *9*9Z@?.?.F6:2 **43 9  9|" K\65" L  L ^1 r