U
    Vc                     @   s8   d Z ddlmZ ddlmZ dZdZdd Zdd Zd	S )
z&Information about histogram summaries.    )summary_pb2)plugin_data_pb2Z
histogramsc                 C   s,   t jtd}tj| |tjjt| ddS )zCreate a `summary_pb2.SummaryMetadata` proto for histogram plugin data.

    Returns:
      A `summary_pb2.SummaryMetadata` protobuf object.
    )version)Zplugin_namecontent)display_nameZsummary_descriptionZplugin_data)r   HistogramPluginDataPROTO_VERSIONr   ZSummaryMetadataZ
PluginDataPLUGIN_NAMEZSerializeToString)r   descriptionr    r   J/tmp/pip-unpacked-wheel-g8kmtpbc/tensorboard/plugins/histogram/metadata.pycreate_summary_metadata   s     r   c                 C   sD   t | tstd| dkr"t S tj| }|jdkr<|S |S dS )zParse summary metadata to a Python object.

    Arguments:
      content: The `content` field of a `SummaryMetadata` proto
        corresponding to the histogram plugin.

    Returns:
      A `HistogramPluginData` protobuf object.
    zContent type must be bytess   {}r   N)
isinstancebytes	TypeErrorr   r   Z
FromStringr   )r   resultr   r   r   parse_plugin_metadata,   s    


r   N)	__doc__Ztensorboard.compat.protor   Ztensorboard.plugins.histogramr   r	   r   r   r   r   r   r   r   <module>   s   