Skip to content

Commit

Permalink
[MINOR][CONNECT] Fix missing request metadata in RPC call
Browse files Browse the repository at this point in the history
### What changes were proposed in this pull request?
The RPC call for fetching the enriched error messages did not receive the GPRC channel per-request metadata, similar to all other other cases of calling the stub. This patch fixes this issue.

### Why are the changes needed?
Compatibility

### Does this PR introduce _any_ user-facing change?
No

### How was this patch tested?
Existing tests.

### Was this patch authored or co-authored using generative AI tooling?
No

Closes apache#48070 from grundprinzip/minor_metadata.

Authored-by: Martin Grund <martin.grund@databricks.com>
Signed-off-by: Hyukjin Kwon <gurwls223@apache.org>
  • Loading branch information
grundprinzip authored and HyukjinKwon committed Sep 11, 2024
1 parent 50955cb commit 44d08cb
Showing 1 changed file with 1 addition and 1 deletion.
2 changes: 1 addition & 1 deletion python/pyspark/sql/connect/client/core.py
Original file line number Diff line number Diff line change
Expand Up @@ -1786,7 +1786,7 @@ def _fetch_enriched_error(self, info: "ErrorInfo") -> Optional[pb2.FetchErrorDet
req.user_context.user_id = self._user_id

try:
return self._stub.FetchErrorDetails(req)
return self._stub.FetchErrorDetails(req, metadata=self._builder.metadata())
except grpc.RpcError:
return None

Expand Down

0 comments on commit 44d08cb

Please sign in to comment.