U
    Vc                     @   s@   d dl Z d dlmZ G dd deZG dd deZdd ZdS )	    N)export_service_pb2c                   @   s   e Zd ZdZdd ZdS )TensorBoardExporterServiceStub3Service for exporting data from TensorBoard.dev.
  c                 C   sR   |j dtjjtjjd| _|j dtjjtjjd| _	|j dtj
jtjjd| _dS )z;Constructor.

    Args:
      channel: A grpc.Channel.
    zA/tensorboard.service.TensorBoardExporterService/StreamExperiments)Zrequest_serializerZresponse_deserializerzD/tensorboard.service.TensorBoardExporterService/StreamExperimentDataz>/tensorboard.service.TensorBoardExporterService/StreamBlobDataN)Zunary_stream;tensorboard_dot_uploader_dot_proto_dot_export__service__pb2StreamExperimentsRequestSerializeToStringStreamExperimentsResponse
FromStringStreamExperimentsStreamExperimentDataRequestStreamExperimentDataResponseStreamExperimentDataStreamBlobDataRequestStreamBlobDataResponseStreamBlobData)selfZchannel r   V/tmp/pip-unpacked-wheel-g8kmtpbc/tensorboard/uploader/proto/export_service_pb2_grpc.py__init__   s    z'TensorBoardExporterServiceStub.__init__N)__name__
__module____qualname____doc__r   r   r   r   r   r      s   r   c                   @   s(   e Zd ZdZdd Zdd Zdd ZdS )	"TensorBoardExporterServiceServicerr   c                 C   s$   | tjj |d tddS )zIStream the experiment_id of all the experiments owned by the caller.
    Method not implemented!NZset_codegrpcZ
StatusCodeZUNIMPLEMENTEDZset_detailsNotImplementedErrorr   requestcontextr   r   r   r
   &   s    
z4TensorBoardExporterServiceServicer.StreamExperimentsc                 C   s$   | tjj |d tddS )z?Stream scalars for all the runs and tags in an experiment.
    r   Nr   r   r   r   r   r   -   s    
z7TensorBoardExporterServiceServicer.StreamExperimentDatac                 C   s$   | tjj |d tddS )z/Stream blob as chunks for a given blob_id.
    r   Nr   r   r   r   r   r   4   s    
z1TensorBoardExporterServiceServicer.StreamBlobDataN)r   r   r   r   r
   r   r   r   r   r   r   r   "   s   r   c                 C   sj   t j| jtjjtjjdt j| jtj	jtj
jdt j| jtjjtjjdd}t d|}||f d S )N)Zrequest_deserializerZresponse_serializer)r
   r   r   z.tensorboard.service.TensorBoardExporterService)r   Zunary_stream_rpc_method_handlerr
   r   r   r	   r   r   r   r   r   r   r   r   Zmethod_handlers_generic_handlerZadd_generic_rpc_handlers)ZservicerserverZrpc_method_handlersZgeneric_handlerr   r   r   0add_TensorBoardExporterServiceServicer_to_server<   s*     r"   )r   Ztensorboard.uploader.protor   r   objectr   r   r"   r   r   r   r   <module>   s   