Skip to content
This repository was archived by the owner on Jan 14, 2024. It is now read-only.

Commit c42c521

Browse files
committed
#32: Fix clean up support
1 parent 469ca7a commit c42c521

File tree

1 file changed

+4
-2
lines changed

1 file changed

+4
-2
lines changed

src/rkd/executor.py

Lines changed: 4 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -43,6 +43,8 @@ def execute(self, declaration: TaskDeclaration, task_num: int, parent: Union[Gro
4343
session_log: bool = os.getenv('RKD_AUDIT_SESSION_LOG', '').lower() in ['true', '1', 'yes']
4444

4545
# 3. execute
46+
temp = TempManager()
47+
4648
try:
4749
io = IO()
4850
io.set_log_level(log_level if log_level else self.io.get_log_level())
@@ -56,7 +58,7 @@ def execute(self, declaration: TaskDeclaration, task_num: int, parent: Union[Gro
5658
target_files=decide_about_target_log_files(self._ctx, log_to_file, session_log, declaration, task_num)):
5759

5860
task = declaration.get_task_to_execute()
59-
task.internal_inject_dependencies(io, self._ctx, self, TempManager())
61+
task.internal_inject_dependencies(io, self._ctx, self, temp)
6062

6163
result = task.execute(
6264
ExecutionContext(
@@ -78,7 +80,7 @@ def execute(self, declaration: TaskDeclaration, task_num: int, parent: Union[Gro
7880
is_exception = True
7981

8082
finally:
81-
TempManager.finally_clean_up()
83+
temp.finally_clean_up()
8284

8385
if result is True:
8486
self._observer.task_succeed(declaration, parent)

0 commit comments

Comments
 (0)