Use correct union type for provider argument (#1840)

fixes #1839
joke2k · Apr 4, 2023 · 5d5c10e · 5d5c10e
1 parent e58414b
commit 5d5c10e
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/faker/generator.py b/faker/generator.py
@@ -1,7 +1,7 @@
 import random as random_module
 import re
 
-from typing import TYPE_CHECKING, Any, Callable, Dict, Hashable, List, Optional
+from typing import TYPE_CHECKING, Any, Callable, Dict, Hashable, List, Optional, Type, Union
 
 from .typing import SeedType
 
@@ -29,7 +29,7 @@ def __init__(self, **config: Dict) -> None:
         self.__config = dict(list(self.__config.items()) + list(config.items()))
         self.__random = random
 
-    def add_provider(self, provider: "BaseProvider") -> None:
+    def add_provider(self, provider: Union["BaseProvider", Type["BaseProvider"]]) -> None:
         if isinstance(provider, type):
             provider = provider(self)