Refactor find_provider function to use Optional type for provider_name and specify return type as Type[BaseProvider]

This commit is contained in:
2024-10-30 17:49:40 -04:00
parent e34f9b106c
commit d972f1cd85
+3 -3
View File
@@ -1,12 +1,12 @@
import difflib
from typing import Union
from typing import Optional, Type
from .providers import providers
from .providers import providers, BaseProvider
_PROVIDER_NAMES = [provider.NAME.lower() for provider in providers]
def find_provider(provider_name: Union[str, None]):
def find_provider(provider_name: Optional[str]) -> Type[BaseProvider]:
"""
Find and instantiate a provider by name.