Skip to content

Commit d69df60

Browse files
psiddhGithub Executorch
authored andcommitted
Fix formatting of docstring in _apply_replace_quant_nodes
1 parent 7a4d15b commit d69df60

1 file changed

Lines changed: 3 additions & 1 deletion

File tree

examples/arm/aot_arm_compiler.py

Lines changed: 3 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -198,13 +198,15 @@ def _load_serialized_model(
198198

199199
return model, example_inputs
200200

201+
201202
def _apply_replace_quant_nodes(edge, args):
202-
""" Apply the replace_quant_nodes pass to the edge graph module. """
203+
"""Apply the replace_quant_nodes pass to the edge graph module."""
203204

204205
if args.target != "vgf" and not args.direct_drive:
205206
edge = edge.transform([ReplaceQuantNodesPass()])
206207
return edge
207208

209+
208210
def get_model_and_inputs_from_name(
209211
model_name: str, model_input: str | None
210212
) -> Tuple[torch.nn.Module, Any]:

0 commit comments

Comments
 (0)