CriteriaDescriptor: Fix typing for enum_class keyword parameter.

Signed-off-by: Chris PeBenito <chpebeni@linux.microsoft.com>
This commit is contained in:
Chris PeBenito 2022-11-30 14:43:41 -05:00
parent ee21412234
commit f1e0035029

View File

@ -18,7 +18,7 @@ from abc import ABC, abstractmethod
from collections import defaultdict
from collections.abc import Collection
from enum import Enum
from typing import Any, Callable, MutableMapping, Optional, Union
from typing import Any, Callable, MutableMapping, Optional, Type, Union
from weakref import WeakKeyDictionary
from .util import validate_perms_any
@ -57,7 +57,7 @@ class CriteriaDescriptor:
def __init__(self, name_regex: Optional[str] = None,
lookup_function: Optional[Union[Callable, str]] = None,
default_value=None, enum_class: Optional[Enum] = None) -> None:
default_value=None, enum_class: Optional[Type[Enum]] = None) -> None:
assert name_regex or lookup_function or enum_class, \
"A simple attribute should be used if there is no regex, lookup function, or enum."