0hdZddlZddlZddlmZiZiZejZedddGddZ ed d d Z ed d ddZ edddZ eddddZ e Zy)z&Python utilities required by TF-Keras.N) keras_exportz keras.saving.custom_object_scopezkeras.utils.custom_object_scopezkeras.utils.CustomObjectScopec"eZdZdZdZdZdZy)CustomObjectScopeaExposes custom classes/functions to TF-Keras deserialization internals. Under a scope `with custom_object_scope(objects_dict)`, TF-Keras methods such as `tf.keras.models.load_model` or `tf.keras.models.model_from_config` will be able to deserialize any custom object referenced by a saved config (e.g. a custom layer or metric). Example: Consider a custom regularizer `my_regularizer`: ```python layer = Dense(3, kernel_regularizer=my_regularizer) # Config contains a reference to `my_regularizer` config = layer.get_config() ... # Later: with custom_object_scope({'my_regularizer': my_regularizer}): layer = Dense.from_config(config) ``` Args: *args: Dictionary or dictionaries of `{name: object}` pairs. c ||_d|_yN)custom_objectsbackup)selfargss a/var/www/html/engine/venv/lib/python3.12/site-packages/tf_keras/src/saving/object_registration.py__init__zCustomObjectScope.__init__<s" ctjj|_|jD]!}tjj |#|Sr)_THREAD_LOCAL_CUSTOM_OBJECTS__dict__copyr rupdate)r objectss r __enter__zCustomObjectScope.__enter__@sH2;;@@B ** BG ( 1 1 8 8 A B rctjjtjj|jyr)rrclearrr )r r kwargss r __exit__zCustomObjectScope.__exit__Fs,$--335$--44T[[ArN)__name__ __module__ __qualname____doc__r rrrr rrs 2 Brrzkeras.saving.get_custom_objectszkeras.utils.get_custom_objectsctS)aRetrieves a live reference to the global dictionary of custom objects. Custom objects set using using `custom_object_scope` are not added to the global dictionary of custom objects, and will not appear in the returned dictionary. Example: ```python get_custom_objects().clear() get_custom_objects()['MyObject'] = MyObject ``` Returns: Global dictionary mapping registered class names to classes. )_GLOBAL_CUSTOM_OBJECTSrrr get_custom_objectsr!Ks ( "!rz(keras.saving.register_keras_serializablez'keras.utils.register_keras_serializablecfd}|S)aRegisters an object with the TF-Keras serialization framework. This decorator injects the decorated class or function into the TF-Keras custom object dictionary, so that it can be serialized and deserialized without needing an entry in the user-provided custom object dict. It also injects a function that TF-Keras will call to get the object's serializable string key. Note that to be serialized and deserialized, classes must implement the `get_config()` method. Functions do not have this requirement. The object will be registered under the key 'package>name' where `name`, defaults to the object name if not passed. Example: ```python # Note that `'my_package'` is used as the `package` argument here, and since # the `name` argument is not provided, `'MyDense'` is used as the `name`. @keras.saving.register_keras_serializable('my_package') class MyDense(keras.layers.Dense): pass assert keras.saving.get_registered_object('my_package>MyDense') == MyDense assert keras.saving.get_registered_name(MyDense) == 'my_package>MyDense' ``` Args: package: The package that this class belongs to. This is used for the `key` (which is `"package>name"`) to idenfify the class. Note that this is the first argument passed into the decorator. name: The name to serialize this class under in this package. If not provided or `None`, the class' name will be used (note that this is the case when the decorator is used with only one argument, which becomes the `package`). Returns: A decorator that registers the decorated class with the passed names. cn |j}dz|z}tj|rt|ds t d|t |<|t |<|S)z get_configzACannot register a class that does not have a get_config() method.)rinspectisclasshasattr ValueErrorr _GLOBAL_CUSTOM_NAMES)arg class_nameregistered_namenamepackages r decoratorz.register_keras_serializable..decoratorse!-T3<< !C-*4 ??3 \(B'  36/$3S! rr)r/r.r0s`` r register_keras_serializabler1bsZ rz keras.saving.get_registered_namezkeras.utils.get_registered_namec<|tvr t|S|jS)aReturns the name registered to an object within the TF-Keras framework. This function is part of the TF-Keras serialization and deserialization framework. It maps objects to the string names associated with those objects for serialization/deserialization. Args: obj: The object to look up. Returns: The name associated with the object, or the default Python name if the object is not registered. )r*r)objs r get_registered_namer4s"" ""#C((||rz"keras.saving.get_registered_objectz!keras.utils.get_registered_objectc|tjvrtj|S|tvr t|S|r ||vr||S|r ||vr||Sy)aReturns the class associated with `name` if it is registered with TF-Keras. This function is part of the TF-Keras serialization and deserialization framework. It maps strings to the objects associated with them for serialization/deserialization. Example: ```python def from_config(cls, config, custom_objects=None): if 'my_custom_object_name' in config: config['hidden_cls'] = tf.keras.saving.get_registered_object( config['my_custom_object_name'], custom_objects=custom_objects) ``` Args: name: The name to look up. custom_objects: A dictionary of custom objects to look the name up in. Generally, custom_objects is provided by the user. module_objects: A dictionary of custom objects to look the name up in. Generally, module_objects is provided by midlevel library implementers. Returns: An instantiable class associated with `name`, or `None` if no such class exists. N)rrr )r.rmodule_objectss r get_registered_objectr7se> +444+44T:: ' '%d++ DN2d## DN2d## r)CustomN)NN)rr& threading tensorflow.python.util.tf_exportrr r*localrrr!r1r4r7custom_object_scoperrr r=s-:.y0&%# &B&B  &BR%'G""(.-9 9x&(I((*M$$P(r