Adds trace info to remote executor's _request.
Intention here is to investigate how much overhead is introduced by the synchronous RPC client.
@@ -178,7 +178,9 @@ class _BidiStream:
self._is_initialized = False
+@tracing.trace(span=True)
def _request(rpc_func, request):
+ """Populates trace context and reraises gRPC errors with retryable info."""
with tracing.wrap_rpc_in_trace_context():
try:
return rpc_func(request)
