From 43df01ecff5e31cccc10b60948d7b74facac917b Mon Sep 17 00:00:00 2001 From: Michael Richard Mckinsey Date: Mon, 7 Aug 2023 16:54:28 -0700 Subject: [PATCH] black --- thicket/ncu.py | 13 +++++++++---- 1 file changed, 9 insertions(+), 4 deletions(-) diff --git a/thicket/ncu.py b/thicket/ncu.py index d4684daf..ed30cc65 100644 --- a/thicket/ncu.py +++ b/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