black

thicket/ncu.py

@@ -101,9 +101,13 @@ def _read_ncu(thicket, ncu_report_mapping):
                 # Grab first action
                 first_action = range.action_by_idx(0)
                 # Metric names
-                metric_names = [first_action[name].name() for name in first_action.metric_names()]
+                metric_names = [
+                    first_action[name].name() for name in first_action.metric_names()
+                ]
                 # Setup rollup dict
-                rollup_dict = {name: first_action[name].rollup_operation() for name in metric_names}
+                rollup_dict = {
+                    name: first_action[name].rollup_operation() for name in metric_names
+                }
 
                 # Query action in range
                 for action in tqdm(range):
@@ -143,9 +147,10 @@ def _read_ncu(thicket, ncu_report_mapping):
                         # Set mapping
                         kernel_map[kernel_name] = matched_node
 
-
                         metric_values = [action[name].value() for name in metric_names]
                         assert len(metric_names) == len(metric_values)
-                        data_dict[(matched_node, ncu_hash)].append(dict(zip(metric_names, metric_values)))
+                        data_dict[(matched_node, ncu_hash)].append(
+                            dict(zip(metric_names, metric_values))
+                        )
 
         return data_dict, rollup_dict