U
    Vc                     @   s$   d Z ddlmZ G dd deZdS )z'Dry-run stubs for various rpc services.    )write_service_pb2c                   @   s8   e Zd ZdZdd Zdd Zdd Zdd	 Zd
d ZdS )DryRunTensorBoardWriterStubzA dry-run TensorBoardWriter gRPC Server.

    Only the methods used by the `tensorboard dev upload` are
    mocked out in this class.

    When additional methods start to be used by the command,
    their mocks should be added to this class.
    c                 K   s   ~~t  S )z9Create a new experiment and remember it has been created.)r   ZCreateExperimentResponseselfrequestkwargs r   F/tmp/pip-unpacked-wheel-g8kmtpbc/tensorboard/uploader/dry_run_stubs.pyCreateExperiment   s    z,DryRunTensorBoardWriterStub.CreateExperimentc                 K   s   ~~t  S N)r   ZWriteScalarResponser   r   r   r	   WriteScalar$   s    z'DryRunTensorBoardWriterStub.WriteScalarc                 K   s   ~~t  S r   )r   ZWriteTensorResponser   r   r   r	   WriteTensor(   s    z'DryRunTensorBoardWriterStub.WriteTensorc                 K   s   ~~t jddS )NZdummy_blob_sequence_id)Zblob_sequence_id)r   ZGetOrCreateBlobSequenceResponser   r   r   r	   GetOrCreateBlobSequence,   s    z3DryRunTensorBoardWriterStub.GetOrCreateBlobSequencec                 k   s   ~|D ]}t  V  qd S r   )r   ZWriteBlobResponse)r   r   r   itemr   r   r	   	WriteBlob2   s    z%DryRunTensorBoardWriterStub.WriteBlobN)	__name__
__module____qualname____doc__r
   r   r   r   r   r   r   r   r	   r      s   	r   N)r   Ztensorboard.uploader.protor   objectr   r   r   r   r	   <module>   s   