@@ -43,10 +43,10 @@ def create_event_dict(start_time, nodes_list):
4343 events = {}
4444 for node in nodes_list :
4545 # 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 )
5050
5151 # Init and format event-based nodes
5252 node ['estimated_threads' ] = estimated_threads
@@ -170,18 +170,21 @@ def calculate_resource_timeseries(events, resource):
170170 and the resource amount as values
171171 '''
172172
173+ # Import packages
174+ import pandas as pd
175+
173176 # Init variables
174177 res = OrderedDict ()
175178 all_res = 0.0
176179
177180 # Iterate through the events
178181 for tdelta , event in sorted (events .items ()):
179182 if event ['event' ] == "start" :
180- if resource in event and event [resource ] != 'Unkown ' :
183+ if resource in event and event [resource ] != 'Unknown ' :
181184 all_res += float (event [resource ])
182185 current_time = event ['start' ];
183186 elif event ['event' ] == "finish" :
184- if resource in event and event [resource ] != 'Unkown ' :
187+ if resource in event and event [resource ] != 'Unknown ' :
185188 all_res -= float (event [resource ])
186189 current_time = event ['finish' ];
187190 res [current_time ] = all_res
0 commit comments