pytorch

716ef2f5 - Improve code to make it more pythonic. (#99720)

Improve code to make it more pythonic. (#99720) No need to use `keys()` method to assert a key is in or not in a dict. Pull Request resolved: https://github.com/pytorch/pytorch/pull/99720 Approved by: https://github.com/kit1980