Skip to content

Commit

Permalink
prettify the code by removing more indent
Browse files Browse the repository at this point in the history
  • Loading branch information
Francesca Wang committed Apr 3, 2024
1 parent 8ac92b1 commit 20cb839
Showing 1 changed file with 1 addition and 1 deletion.
2 changes: 1 addition & 1 deletion torchfix/visitors/vision/models_import.py
Original file line number Diff line number Diff line change
Expand Up @@ -16,7 +16,7 @@ def visit_Import(self, node: cst.Import) -> None:
for imported_item in node.names:
if m.matches(imported_item, m.ImportAlias(
name=m.Attribute(value=m.Name("torchvision"),
attr=m.Name("models")),
attr=m.Name("models")),
asname=m.AsName(name=m.Name("models"))
)):
# Replace only if the import statement has no other names
Expand Down

0 comments on commit 20cb839

Please sign in to comment.