After download is complete remove the remote job dir

i-VRESSE · Mar 8, 2024 · 5b3bc23 · 5b3bc23
1 parent 565e23b
commit 5b3bc23
Show file tree

Hide file tree

Showing 2 changed files with 1 addition and 3 deletions.
diff --git a/src/bartender/filesystems/abstract.py b/src/bartender/filesystems/abstract.py
@@ -48,8 +48,6 @@ async def delete(self, description: JobDescription) -> None:
         Args:
             description: Remote directory to delete.
         """
-        # after download or cancellation you might want to
-        # delete the remote job directory
 
     async def close(self) -> None:
         """Close filesystem."""
diff --git a/src/bartender/filesystems/queue.py b/src/bartender/filesystems/queue.py
@@ -48,7 +48,7 @@ async def perform_download(
 
     await filesystem.download(localized_description, description)
 
-    # TODO for non-local file system should also remove remote files?
+    await filesystem.delete(localized_description)
 
     if job_id is not None:
         await job_dao.update_job_state(job_id, state)