wrapper/construct-typing
1
0
Fork 0
Code Issues Pull requests Projects Releases Packages Wiki Activity Actions

fixed error message from EnumBase and FlagsEnumBase that occures since pyright v1.1.320

Browse source
This commit is contained in:
Tim Riddermann 2023-08-03 09:21:30 +02:00
parent af98d2004d
commit c405d09d5f
1 changed files with 4 additions and 2 deletions
Download patch file Download diff file Expand all files Collapse all files

6
construct_typed/tenum.py
View file

@ -1,6 +1,8 @@
import enum
import typing as t
from typing_extensions import Self
from .generic_wrapper import *
@ -45,7 +47,7 @@ class EnumBase(enum.IntEnum):
'This is the running state.'
"""
def __new__(cls, val: t.Union[EnumValue, int]) -> "EnumBase":
def __new__(cls, val: t.Union[EnumValue, int]) -> "Self":
if isinstance(val, EnumValue):
obj = int.__new__(cls, val.value)
obj._value_ = val.value
@ -158,7 +160,7 @@ class FlagsEnumBase(enum.IntFlag):
'This is option two.'
"""
def __new__(cls, val: t.Union[EnumValue, int]) -> "FlagsEnumBase":
def __new__(cls, val: t.Union[EnumValue, int]) -> "Self":
if isinstance(val, EnumValue):
obj = int.__new__(cls, val.value)
obj._value_ = val.value