Fix type check error

layday · Jan 15, 2024 · aab34b8 · aab34b8
1 parent febe14c
commit aab34b8
Show file tree

Hide file tree

Showing 2 changed files with 4 additions and 4 deletions.
diff --git a/src/instawow/common.py b/src/instawow/common.py
@@ -19,7 +19,7 @@ class _FlavourKeyedEnum(Protocol[_TEnum]):  # pragma: no cover
     VanillaClassic: _TEnum
     Classic: _TEnum
 
-    def __getitem__(self, __key: str) -> _TEnum:
+    def __class_getitem__(cls, __key: str) -> _TEnum:
         ...
 
 
@@ -38,8 +38,8 @@ class Flavour(StrEnum):
     def from_flavour_keyed_enum(cls, flavour_keyed_enum: enum.Enum) -> Self:
         return cls[flavour_keyed_enum.name]
 
-    def to_flavour_keyed_enum(self, flavour_keyed_enum: _FlavourKeyedEnum[_TEnum]) -> _TEnum:
-        return flavour_keyed_enum[self.name]
+    def to_flavour_keyed_enum(self, flavour_keyed_enum: type[_FlavourKeyedEnum[_TEnum]]) -> _TEnum:
+        return flavour_keyed_enum[self.name]  # pyright: ignore[reportGeneralTypeIssues]
 
 
 class FlavourVersionRange(enum.Enum):

diff --git a/src/instawow/resolvers.py b/src/instawow/resolvers.py
@@ -159,5 +159,5 @@ async def catalogue(cls, web_client: http.ClientSessionType) -> AsyncIterator[Ca
         yield
 
 
-class _DummyResolver(Resolver):  # pyright: ignore[reportGeneralTypeIssues]
+class _DummyResolver(Resolver):
     pass
-Original file line number
+Diff line change
@@ Expand Up @@
             yield
-    class _DummyResolver(Resolver):  # pyright: ignore[reportGeneralTypeIssues]
+    class _DummyResolver(Resolver):
         pass