Skip to content

Commit 580c6d7

Browse files
committed
Removed more REST API calls
1 parent 1f3609a commit 580c6d7

3 files changed

Lines changed: 20 additions & 10 deletions

File tree

examples/json_workflow_simulator/simulator.py

Lines changed: 1 addition & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -176,6 +176,7 @@ def main():
176176
for f in files:
177177
ss.create_file_copy(f)
178178

179+
179180
# We are now ready to schedule the workflow
180181
print(f"Starting my main loop!")
181182
while not workflow.is_done():

wrench/simulation.py

Lines changed: 14 additions & 7 deletions
Original file line numberDiff line numberDiff line change
@@ -517,22 +517,29 @@ def create_workflow_from_json(self, json_object: json, reference_flop_rate: str,
517517
# Create the workflow
518518
workflow = Workflow(self, response["workflow_name"])
519519

520+
521+
# Create the files (caching parameter values)
522+
for file_spec in response["files"]:
523+
file_name = file_spec["name"]
524+
file_size = file_spec["size"]
525+
self.files[file_name] = File(self, file_name, file_size)
526+
520527
# Create the tasks (caching parameter values)
521528
for task_spec in response["tasks"]:
522529
task_name = task_spec["name"]
523530
task_flops = task_spec["flops"]
524531
task_min_num_cores = task_spec["min_num_cores"]
525532
task_max_num_cores = task_spec["max_num_cores"]
526533
task_memory = task_spec["memory"]
527-
workflow.tasks[task_name] = Task(self, workflow, task_name, task_flops, task_min_num_cores, task_max_num_cores, task_memory)
534+
input_file_list = []
535+
for file_name in task_spec["input_files"]:
536+
input_file_list.append(self.files[file_name])
537+
output_file_list = []
538+
for file_name in task_spec["output_files"]:
539+
output_file_list.append(self.files[file_name])
528540

541+
workflow.tasks[task_name] = Task(self, workflow, task_name, task_flops, task_min_num_cores, task_max_num_cores, task_memory, input_file_list, output_file_list)
529542

530-
# Create the files (caching parameter values)
531-
print(response["files"])
532-
for file_spec in response["files"]:
533-
file_name = file_spec["name"]
534-
file_size = file_spec["size"]
535-
self.files[file_name] = File(self, file_name, file_size)
536543

537544
return workflow
538545

wrench/task.py

Lines changed: 5 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -32,7 +32,9 @@ def __init__(self, simulation: Simulation, workflow: Workflow, name: str,
3232
flops: flops = None,
3333
min_num_cores: number = None,
3434
max_num_cores: number = None,
35-
memory: number = None) -> None:
35+
memory: number = None,
36+
input_files: List[File] = None,
37+
output_files: List[File] = None) -> None:
3638
"""
3739
Constructor
3840
:param simulation: simulation object
@@ -48,8 +50,8 @@ def __init__(self, simulation: Simulation, workflow: Workflow, name: str,
4850
self.min_num_cores = min_num_cores
4951
self.max_num_cores = max_num_cores
5052
self.memory = memory
51-
self.input_files = None
52-
self.output_files = None
53+
self.input_files = input_files
54+
self.output_files = output_files
5355

5456
class TaskState(Enum):
5557
NOT_READY = 0

0 commit comments

Comments
 (0)