Skip to content

Commit ad85333

Browse files
committed
DEV-22328: mypy fix
1 parent cdaa167 commit ad85333

File tree

1 file changed

+6
-3
lines changed

1 file changed

+6
-3
lines changed

huntflow_api_client/utils.py

Lines changed: 6 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -1,9 +1,12 @@
11
from enum import Enum
2+
from typing import Any, Callable, Dict, Type, TypeVar
23

4+
T = TypeVar("T", bound=Enum)
35

4-
def extend_enum(inherited_enum):
5-
def wrapper(added_enum):
6-
joined = {}
6+
7+
def extend_enum(inherited_enum: Type[T]) -> Callable:
8+
def wrapper(added_enum: Type[T]) -> Enum:
9+
joined: Dict[str, Any] = {}
710
for item in inherited_enum:
811
joined[item.name] = item.value
912
for item in added_enum:

0 commit comments

Comments
 (0)