@@ -43,10 +43,10 @@ def create_event_dict(start_time, nodes_list):
43
43
events = {}
44
44
for node in nodes_list :
45
45
# Format node fields
46
- estimated_threads = int ( node .get ('num_threads' ) , 1 )
47
- estimated_memory_gb = float ( node .get ('estimated_memory_gb' , 1.0 ) )
48
- runtime_threads = int ( node .get ('runtime_threads' ) , 0 )
49
- runtime_memory_gb = float ( node .get ('runtime_memory_gb' , 0.0 ) )
46
+ estimated_threads = node .get ('num_threads' , 1 )
47
+ estimated_memory_gb = node .get ('estimated_memory_gb' , 1.0 )
48
+ runtime_threads = node .get ('runtime_threads' , 0 )
49
+ runtime_memory_gb = node .get ('runtime_memory_gb' , 0.0 )
50
50
51
51
# Init and format event-based nodes
52
52
node ['estimated_threads' ] = estimated_threads
@@ -170,18 +170,21 @@ def calculate_resource_timeseries(events, resource):
170
170
and the resource amount as values
171
171
'''
172
172
173
+ # Import packages
174
+ import pandas as pd
175
+
173
176
# Init variables
174
177
res = OrderedDict ()
175
178
all_res = 0.0
176
179
177
180
# Iterate through the events
178
181
for tdelta , event in sorted (events .items ()):
179
182
if event ['event' ] == "start" :
180
- if resource in event and event [resource ] != 'Unkown ' :
183
+ if resource in event and event [resource ] != 'Unknown ' :
181
184
all_res += float (event [resource ])
182
185
current_time = event ['start' ];
183
186
elif event ['event' ] == "finish" :
184
- if resource in event and event [resource ] != 'Unkown ' :
187
+ if resource in event and event [resource ] != 'Unknown ' :
185
188
all_res -= float (event [resource ])
186
189
current_time = event ['finish' ];
187
190
res [current_time ] = all_res
0 commit comments