Short form of enum.StrEnum fails

[dave-shawley]

Pyre Bug

Bug description

Similar to #698 only using enum.StrEnum in place of enum.Enum.

Reproduction steps

import enum

Color = enum.StrEnum('Color', ['Red', 'Blue'])
print(Color.Red)

Expected behavior

Type checking works.

Logs
Please include any relevant logs here:

ƛ No watchman binary found.
To enable pyre incremental, you can install watchman: https://facebook.github.io/watchman/docs/install
ƛ Defaulting to non-incremental check.
ƛ Found 2 type errors!
checkit.py:3:8 Too many arguments [19]: Call `enum.StrEnum.__new__` expects 1 positional argument, 2 were provided.
checkit.py:4:6 Undefined attribute [16]: `enum.StrEnum` has no attribute `Red`.
zsh: exit 1     pyre

Please run your reproduction steps followed by pyre rage > pyre_rage.log, and upload the file here:

pyre_rage.log

Additional context

Using hatch to manage virtual environments and have pyre installed into the environment.

[yangdanny97]

Thanks for reporting this issue, I'll take a look next week

[yangdanny97]

I can confirm that Pyre doesn't support functional syntax for any enums besides enum.Enum.

I could add support for StrEnum and IntEnum fairly quickly, but the current setup would need major changes to have it work for any enum subclass. So a general solution would need to wait till we're done making some architectural changes to the system.

[dave-shawley]

@yangdanny97 if you are in the process of refactoring, I would wait to be honest. Especially since switching to the class syntax solves the issue. Thanks for the great tool!