microsoft/pyright
1
1
Fork 0
mirror of https://github.com/microsoft/pyright.git synced 2024-10-10 14:59:42 +03:00
Code Issues Projects Releases Wiki Activity

Fixed regression that results in the incorrect type evaluation for the value attribute of a StrEnum or IntEnum class. This addresses #7983.

Browse Source
This commit is contained in:
Eric Traut 2024-05-23 09:18:17 -07:00
parent bfc60526a0
commit aed4c75980
2 changed files with 4 additions and 3 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

5
packages/pyright-internal/src/analyzer/enums.ts
View File

@ -587,9 +587,10 @@ export function getTypeOfEnumMember(
if (memberName === 'value' || memberName === '_value_') {
// Does the class explicitly override this member? Or it it using the
// standard behavior provided by the "Enum" class?
// standard behavior provided by the "Enum" class and other built-in
// subclasses like "StrEnum" and "IntEnum"?
const memberInfo = lookUpClassMember(classType, memberName);
if (memberInfo && isClass(memberInfo.classType) && !ClassType.isBuiltIn(memberInfo.classType, 'Enum')) {
if (memberInfo && isClass(memberInfo.classType) && !ClassType.isBuiltIn(memberInfo.classType)) {
return undefined;
}

2
packages/pyright-internal/src/tests/samples/enum1.py
View File

@ -217,7 +217,7 @@ reveal_type(te15_A.value, expected_text="str")
reveal_type(te15_A._value_, expected_text="str")
class TestEnum16(Enum):
class TestEnum16(IntEnum):
A = 1
B = 2
C = 3