Refactor how import and call updates are handled

pytorch-labs / torchfix

TorchFix - a linter for PyTorch-using code with autofix support

Other

98 stars 16 forks source link

Refactor how import and call updates are handled #53

Closed kit1980 closed 6 months ago

kit1980 commented 6 months ago

This PR refactors code to update calls and imports from TorchNonPublicAliasVisitor into common functionality and uses the functionality for TorchDeprecatedSymbolsVisitor.

This allowed to fix https://github.com/pytorch-labs/torchfix/issues/50 and resolve a TODO to remove _UpdateFunctorchImports.