eskarimov commented on a change in pull request #19736:
URL: https://github.com/apache/airflow/pull/19736#discussion_r754526457
##########
File path: airflow/providers/databricks/hooks/databricks.py
##########
@@ -351,31 +362,31 @@ def _do_api_call(self, endpoint_info, json):
def _log_request_error(self, attempt_num: int, error: str) -> None:
self.log.error('Attempt %s API Request to Databricks failed with
reason: %s', attempt_num, error)
- def run_now(self, json: dict) -> str:
+ def run_now(self, json: dict) -> int:
"""
Utility function to call the ``api/2.0/jobs/run-now`` endpoint.
:param json: The data used in the body of the request to the
``run-now`` endpoint.
:type json: dict
:return: the run_id as a string
- :rtype: str
+ :rtype: int
"""
response = self._do_api_call(RUN_NOW_ENDPOINT, json)
return response['run_id']
- def submit_run(self, json: dict) -> str:
+ def submit_run(self, json: dict) -> int:
Review comment:
I was just referring to the Databricks API docs, where it's int.
I think should be handled correctly, at least I've checked to align types
everywhere.
--
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]
