Skip to content

Commit becb19a

Browse files
Fixed formatting
1 parent fe31c6c commit becb19a

File tree

2 files changed

+11
-3
lines changed

2 files changed

+11
-3
lines changed

devtools/inspector/_inspector.py

Lines changed: 8 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -1065,7 +1065,11 @@ def __init__(
10651065
self._enable_module_hierarchy = enable_module_hierarchy
10661066
self._consume_etrecord(module_name, method_name)
10671067

1068-
def _consume_etrecord(self, module_name: Optional[str] = None, method_name: Optional[str] = None,) -> None:
1068+
def _consume_etrecord(
1069+
self,
1070+
module_name: Optional[str] = None,
1071+
method_name: Optional[str] = None,
1072+
) -> None:
10691073
"""
10701074
If an ETRecord is provided, connect it to the EventBlocks and populate the Event metadata.
10711075
@@ -1089,7 +1093,9 @@ def _consume_etrecord(self, module_name: Optional[str] = None, method_name: Opt
10891093
method_name = FORWARD
10901094
edge_dialect_graph_key = EDGE_DIALECT_GRAPH_KEY
10911095
elif method_name is None or module_name is None:
1092-
raise ValueError("Either both method_name and module_name should be provided or neither should be provided")
1096+
raise ValueError(
1097+
"Either both method_name and module_name should be provided or neither should be provided"
1098+
)
10931099
else:
10941100
method_name = method_name
10951101
edge_dialect_graph_key = f"{module_name}/{method_name}"

devtools/inspector/_inspector_utils.py

Lines changed: 3 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -236,7 +236,9 @@ def is_debug_output(value: Value) -> bool:
236236

237237

238238
def gen_graphs_from_etrecord(
239-
etrecord: ETRecord, enable_module_hierarchy: bool = False, edge_dialect_graph_key: str = EDGE_DIALECT_GRAPH_KEY
239+
etrecord: ETRecord,
240+
enable_module_hierarchy: bool = False,
241+
edge_dialect_graph_key: str = EDGE_DIALECT_GRAPH_KEY,
240242
) -> Mapping[str, OperatorGraph]:
241243
op_graph_map = {}
242244
if etrecord.graph_map is not None:

0 commit comments

Comments
 (0)