Lee-W commented on code in PR #55072:
URL: https://github.com/apache/airflow/pull/55072#discussion_r2318444733
##########
providers/apache/livy/src/airflow/providers/apache/livy/hooks/livy.py:
##########
@@ -559,7 +559,7 @@ async def _do_api_call_async(
url,
json=data if self.method in ("POST", "PATCH") else None,
params=data if self.method == "GET" else None,
- headers=headers,
+ headers=_headers or None,
Review Comment:
I think `heaers=_headers` is the right change
##########
providers/apache/livy/src/airflow/providers/apache/livy/hooks/livy.py:
##########
@@ -629,7 +629,10 @@ async def get_batch_state(self, session_id: int | str) ->
Any:
"""
self._validate_session_id(session_id)
self.log.info("Fetching info for batch session %s", session_id)
- result = await
self.run_method(endpoint=f"{self.endpoint_prefix}/batches/{session_id}/state")
+ result = await self.run_method(
Review Comment:
I think we need to do the same thing for `get_batch_logs`
--
This is an automated message from the Apache Git Service.
To respond to the message, please log on to GitHub and use the
URL above to go to the specific comment.
To unsubscribe, e-mail: [email protected]
For queries about this service, please contact Infrastructure at:
[email protected]
