File tree 1 file changed +9
-4
lines changed
1 file changed +9
-4
lines changed Original file line number Diff line number Diff line change @@ -67,7 +67,11 @@ def fix(a):
67
67
for c in a ["children" ]:
68
68
fix (c )
69
69
70
- fix (new_message ["utilization" ]["rootNode" ]) if new_message ["utilization" ].get ('rootNode' , None ) else None
70
+ (
71
+ fix (new_message ["utilization" ]["rootNode" ])
72
+ if new_message ["utilization" ].get ("rootNode" , None )
73
+ else None
74
+ )
71
75
try :
72
76
new_message ["utilization" ]["tensor_core_usage" ] = message ["utilization" ][
73
77
"tensor_utilization"
@@ -143,11 +147,12 @@ def fix_components(m):
143
147
"components" : message ["energy" ]["components" ],
144
148
"batch_size" : 48 ,
145
149
},
146
- "past_measurements" : message ["energy" ][ "past_measurements" ] ,
150
+ "past_measurements" : message ["energy" ]. get ( "past_measurements" , None ) ,
147
151
}
148
152
149
153
fix_components (new_message ["energy" ]["current" ])
150
- for m in new_message ["energy" ]["past_measurements" ]:
151
- fix_components (m )
154
+ if new_message ["energy" ].get ("past_measurements" , None ):
155
+ for m in new_message ["energy" ]["past_measurements" ]:
156
+ fix_components (m )
152
157
153
158
return new_message
You can’t perform that action at this time.
0 commit comments