Skip to content

Commit cc27bab

Browse files
committed
small fixes
1 parent 8805fce commit cc27bab

3 files changed

Lines changed: 7 additions & 7 deletions

File tree

workflowhub/common/machine.py

Lines changed: 3 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -74,9 +74,9 @@ def __init__(self,
7474
self.hashcode = hashcode
7575

7676
self.cpu_cores: int = cpu['count']
77-
self.cpu_speed: int = cpu['speed']
78-
self.cpu_flops: int = cpu['count'] * cpu['speed'] * 10 ^ 6
79-
self.cpu_vendor: str = cpu['vendor']
77+
self.cpu_speed: int = cpu['speed'] if 'speed' in cpu else 0
78+
self.cpu_flops: int = cpu['count'] * cpu['speed'] * 10 ^ 6 if 'speed' in cpu else 0
79+
self.cpu_vendor: str = cpu['vendor'] if 'vendor' in cpu else None
8080

8181
self.logger.debug("created machine: {0} with {1} cores and {2} FLOPS.".format(
8282
self.name, self.cpu_cores, self.cpu_flops)

workflowhub/trace/logs/makeflow.py

Lines changed: 3 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -118,7 +118,7 @@ def _parse_workflow_file(self):
118118

119119
elif len(line.strip()) > 0:
120120
# task execution command
121-
prefix = line.split()[1 if 'LOCAL' in line else 0].replace('./', '')
121+
prefix = line.replace('./', '').replace('perl', '').strip().split()[1 if 'LOCAL' in line else 0]
122122
task_name = "{}_ID{:06d}".format(prefix, task_id_counter)
123123
task_id_counter += 1
124124

@@ -128,7 +128,7 @@ def _parse_workflow_file(self):
128128
list_files.extend(self._create_files(inputs, FileLink.INPUT, task_name))
129129

130130
# create task
131-
args = line.replace('LOCAL', '').strip()
131+
args = ' '.join(line.replace('LOCAL', '').replace('perl', '').strip().split())
132132
task = Task(name=task_name,
133133
task_type=TaskType.COMPUTE,
134134
runtime=0,
@@ -210,7 +210,7 @@ def _parse_resource_monitor_logs(self):
210210
data = json.load(f)
211211

212212
# task
213-
task = self.args_map[data['command']]
213+
task = self.args_map[data['command'].replace('perl', '').strip()]
214214
task.runtime = float(data['wall_time'][0])
215215
task.cores = int(data['cores'][0])
216216
task.memory = int(data['memory'][0]) * 1000 # MB to KB

workflowhub/trace/trace.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -88,7 +88,7 @@ def __init__(self, input_trace: str, schema_file: Optional[str] = None, logger:
8888
machine['nodeName']: Machine(
8989
name=machine['nodeName'],
9090
cpu={k: v for k, v in machine['cpu'].items()},
91-
system=MachineSystem(machine.get('system', None)),
91+
system=MachineSystem(machine.get('system', None)) if machine.get('system', None) else None,
9292
architecture=machine.get('architecture', None),
9393
memory=machine.get('memory', None),
9494
release=machine.get('release', None),

0 commit comments

Comments
 (0)