Add missing cancel_all_runs function to JobsService (#542)

databricks · Aug 18, 2022 · 4197d5b · 4197d5b
1 parent cdf917f
commit 4197d5b
Showing 1 changed file with 8 additions and 0 deletions.
diff --git 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: