Skip to content

Commit 2e43e8c

Browse files
committed
Storing input/output file list inside task
1 parent f413684 commit 2e43e8c

File tree

1 file changed

+10
-4
lines changed

1 file changed

+10
-4
lines changed

wrench/task.py

Lines changed: 10 additions & 4 deletions
Original file line numberDiff line numberDiff line change
@@ -44,6 +44,8 @@ def __init__(self, simulation: Simulation, workflow: Workflow, name: str) -> Non
4444
self.min_num_cores = None
4545
self.max_num_cores = None
4646
self.memory = None
47+
self.input_files = []
48+
self.output_files = []
4749

4850
class TaskState(Enum):
4951
NOT_READY = 0
@@ -71,31 +73,35 @@ def add_input_file(self, file: File) -> None:
7173
:param file: File name
7274
:type file: File
7375
"""
74-
return self._simulation._add_input_file(self, file)
76+
self._simulation._add_input_file(self, file)
77+
self.input_files.append(file)
7578

7679
def add_output_file(self, file: File) -> None:
7780
"""
7881
Add a file as output file for this task
7982
:param file: File name
8083
:type file: File
8184
"""
82-
return self._simulation._add_output_file(self, file)
85+
self._simulation._add_output_file(self, file)
86+
self.output_files.append(file)
8387

8488
def get_input_files(self) -> List[File]:
8589
"""
8690
Get the list of input files for this task
8791
:return: List of input file names
8892
:rtype: List[File]
8993
"""
90-
return self._simulation._get_task_input_files(self)
94+
return self.input_files
95+
# return self._simulation._get_task_input_files(self)
9196

9297
def get_output_files(self) -> List[File]:
9398
"""
9499
Get the list of output files for this task
95100
:return: List of output file names
96101
:rtype: List[File]
97102
"""
98-
return self._simulation._get_task_output_files(self)
103+
return self.output_files
104+
# return self._simulation._get_task_output_files(self)
99105

100106
def get_flops(self) -> float:
101107
"""

0 commit comments

Comments
 (0)