diff --git a/tensorbay/client/job.py b/tensorbay/client/job.py index a080358c8..4f97878f7 100644 --- a/tensorbay/client/job.py +++ b/tensorbay/client/job.py @@ -165,11 +165,15 @@ def update(self, until_complete: bool = False) -> None: def abort(self) -> None: """Abort a :class:`Job`.""" - self._client.open_api_do("POST", f"jobs/{self.job_id}/abort", self._dataset_id) + self._client.open_api_do( + "POST", f"jobs/{self.job_id}/abort?jobType={self._job_type}", self._dataset_id + ) def retry(self) -> None: """Retry a :class:`Job`.""" - self._client.open_api_do("POST", f"jobs/{self.job_id}/retry", self._dataset_id) + self._client.open_api_do( + "POST", f"jobs/{self.job_id}/retry?jobType={self._job_type}", self._dataset_id + ) class SquashAndMergeJob(Job): diff --git a/tensorbay/client/version.py b/tensorbay/client/version.py index e4b3b176e..eeca694e4 100644 --- a/tensorbay/client/version.py +++ b/tensorbay/client/version.py @@ -627,7 +627,9 @@ def delete_job(self, job_id: str) -> None: job_id: The Job id. """ - self._client.open_api_do("DELETE", f"jobs/{job_id}", self._dataset_id) + self._client.open_api_do( + "DELETE", f"jobs/{job_id}?jobType={self._JOB_TYPE}", self._dataset_id + ) class SquashAndMerge(JobMixin):