Add mypy typing to registration dict.

robertwb · robertwb · commit 051872e6deba · 2023-08-09T16:02:10.000-07:00
diff --git a/sdks/python/apache_beam/yaml/yaml_provider.py b/sdks/python/apache_beam/yaml/yaml_provider.py
@@ -29,6 +29,7 @@
 import uuid
 from typing import Any
 from typing import Callable
+from typing import Dict
 from typing import Iterable
 from typing import Mapping
 
@@ -106,7 +107,7 @@ def as_provider_list(name, lst):
 
 class ExternalProvider(Provider):
   """A Provider implemented via the cross language transform service."""
-  _provider_types = {}
+  _provider_types: Dict[str, Callable[..., Provider]] = {}
 
   def __init__(self, urns, service):
     self._urns = urns
