o
    ?eI                     @   sN   d Z ddlZddlZG dd dejZe Zejdd Zdd Zd	d
 Z	dS )z Context for building SavedModel.    Nc                       s@   e Zd ZdZ fddZdd Zdd Zdd	 Zd
d Z  Z	S )SaveContextz-A context for building a graph of SavedModel.c                    s   t t|   d| _d | _d S NF)superr   __init___in_save_context_optionsself	__class__ k/home/www/facesmatcher.com/pyenv/lib/python3.10/site-packages/tensorflow/python/saved_model/save_context.pyr      s   
zSaveContext.__init__c                 C   s   |   std| jS )NzNot in a SaveContext.)in_save_context
ValueErrorr   r   r   r   r   options   s   zSaveContext.optionsc                 C   s   d| _ || _d S )NTr   r   )r	   r   r   r   r   enter_save_context"      
zSaveContext.enter_save_contextc                 C   s   d| _ d | _d S r   r   r   r   r   r   exit_save_context&   r   zSaveContext.exit_save_contextc                 C   s   | j S )N)r   r   r   r   r   r   *   s   zSaveContext.in_save_context)
__name__
__module____qualname____doc__r   r   r   r   r   __classcell__r   r   r
   r   r      s    r   c                 c   s:    t  rtdt|  z
d V  W t  d S t  w )NzAlready in a SaveContext.)r   r   _save_contextr   r   )r   r   r   r   save_context0   s   
r   c                   C      t  S )z%Returns whether under a save context.)r   r   r   r   r   r   r   ;      r   c                   C   r   )z1Returns the save options if under a save context.)r   r   r   r   r   r   get_save_options@   r   r   )
r   
contextlib	threadinglocalr   r   contextmanagerr   r   r   r   r   r   r   <module>   s   

