Add rule for deprecated _register_pytree_node

pytorch-labs / torchfix

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

Other

98 stars 16 forks source link

Closed ZainRizvi closed 6 months ago

ZainRizvi commented 6 months ago

Testing:

Added unit tests
Ran torchfix against a directory that used this deprecated function and verified that the func was only caught after the change