o ?eU @s@dZddlZddlmZddlmZdZdZGddde Z dS) zRegistry mechanism for "registering" classes/functions for general use. This is typically used with a decorator that calls Register for adding a class or function to a registry. N) tf_logging)compatlocationtypec@s:eZdZdZddgZddZd ddZd d Zd d ZdS)Registryz'Provides a registry for saving objects._name _registrycCs||_i|_dS)zCreates a new registry.N)rrselfnamer e/home/www/facesmatcher.com/pyenv/lib/python3.10/site-packages/tensorflow/python/framework/registry.py__init__&s zRegistry.__init__NcCs|s|j}||jvr |j|t}td|j||j|j|jft dd|||jt j dd}t dt |d}|dkrB||}nd}t|t|i|j|<d S) aERegisters a Python object "candidate" for the given "name". Args: candidate: The candidate object to add to the registry. name: An optional string specifying the registry key for the candidate. If None, candidate.__name__ will be used. Raises: KeyError: If same name is used twice. zJRegistering two %s with name '%s'! (Previous registration was in %s %s:%d)zRegistering %s (%s) in %s.)limitr)UNKNOWNrrrrN)__name__r _LOCATION_TAGKeyErrorrr filenamelinenologgingZvlog traceback extract_stackminlen _TYPE_TAG)r candidater framestackZ stack_indexZ location_tagr r r register+s     zRegistry.registercCs |jS)zWLists registered items. Returns: A list of names of registered objects. )rkeys)r r r r listIs z Registry.listcCs4t|}||jvr|j|tStd|j|f)zLooks up "name". Args: name: a string specifying the registry key for the candidate. Returns: Registered object if found Raises: LookupError: if "name" has not been registered. z %s registry has no entry for: %s)rZas_strrr LookupErrorrr r r r lookupQs  zRegistry.lookup)N) r __module__ __qualname____doc__ __slots__rr"r$r&r r r r r!s  r) r)rZtensorflow.python.platformrrZtensorflow.python.utilrrrobjectrr r r r s