def _new_request(self):
"""Allocates a new request and refreshes the budget."""
self._request = write_service_pb2.WriteTensorRequest()
self._runs.clear()
self._tags.clear()
self._request.experiment_id = self._experiment_id
self._byte_budget_manager.reset(self._request)
def testWriteTensor(self):
self._stub.WriteTensor(write_service_pb2.WriteTensorRequest())
