Skip to content

Commit fb86cd0

Browse files
authored
Add cancel to docker executor (#233)
* Add cancel to docker executor Signed-off-by: Hemil Desai <[email protected]> * fix Signed-off-by: Hemil Desai <[email protected]> --------- Signed-off-by: Hemil Desai <[email protected]>
1 parent f09ddb1 commit fb86cd0

File tree

1 file changed

+7
-0
lines changed
  • nemo_run/run/torchx_backend/schedulers

1 file changed

+7
-0
lines changed

nemo_run/run/torchx_backend/schedulers/docker.py

Lines changed: 7 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -254,6 +254,13 @@ def local_logs(container: DockerContainer):
254254
else:
255255
return logs
256256

257+
def _cancel_existing(self, app_id: str) -> None:
258+
req = DockerJobRequest.load(app_id=app_id)
259+
if not req:
260+
return None
261+
for container in req.containers:
262+
container.delete(client=self._docker_client, id=req.id)
263+
257264
def close(self) -> None:
258265
# terminate all apps
259266
for req in self._scheduled_reqs:

0 commit comments

Comments
 (0)