torch.Tag doesn't have accurate mypy info

[ezyang]

🐛 Describe the bug

  Error (MYPY) [attr-defined]
    Module has no attribute "Tag"

        151  |        r = func_overload.decompose(*args, **kwargs)
        152  |        if r is not NotImplemented:
        153  |            return r
    >>> 154  |    if torch.Tag.data_dependent_output in func_overload.tags:
        155  |        # Check if all of the Tensor inputs are constants
        156  |        all_constant = True
        157  |

Versions

master

cc @ezyang @malfet @rgommers @xuzhao9 @gramster

[malfet]

What module is it part of? And what's your mypy version

[ezyang]

it's a pybind11 enum codegen from torch._C

[eellison]

Any update on this ? There are a ton of # type: ignore[attr-defined] being added everywhere