pytorch

3d28ab07 - Minor follow up fixes for python registration

Minor follow up fixes for python registration Pull Request resolved: https://github.com/pytorch/pytorch/pull/76892 Approved by: https://github.com/albanD