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())