diff --git a/databricks_cli/sdk/service.py b/databricks_cli/sdk/service.py index f71eaa7b..ea7871b8 100755 --- a/databricks_cli/sdk/service.py +++ b/databricks_cli/sdk/service.py @@ -392,6 +392,14 @@ def cancel_run(self, run_id, headers=None, version=None): 'POST', '/jobs/runs/cancel', data=_data, headers=headers, version=version ) + def cancel_all_runs(self, job_id, headers=None, version=None): + _data = {} + if job_id is not None: + _data['job_id'] = job_id + return self.client.perform_query( + 'POST', '/jobs/runs/cancel-all', data=_data, headers=headers, version=version + ) + def get_run_output(self, run_id, headers=None, version=None): _data = {} if run_id is not None: