Added Type[BaseModel] Annotation to TypeConfiguration (#163)

Co-authored-by: Hippolyte HENRY <[email protected]>

Author: ammokhov

src/cloudformation_cli_python_lib/resource.py

@@ -60,12 +60,12 @@ def __init__(
         self,
         type_name: str,
         resouce_model_cls: Type[BaseModel],
-        type_configuration_model_cls: Optional[BaseModel] = None,
+        type_configuration_model_cls: Optional[Type[BaseModel]] = None,
     ) -> None:
         self.type_name = type_name
         self._model_cls: Type[BaseModel] = resouce_model_cls
         self._type_configuration_model_cls: Optional[
-            BaseModel
+            Type[BaseModel]
         ] = type_configuration_model_cls
         self._handlers: MutableMapping[Action, HandlerSignature] = {}
 

src/cloudformation_cli_python_lib/utils.py

@@ -140,7 +140,7 @@ class UnmodelledRequest:
     def to_modelled(
         self,
         model_cls: Type[BaseModel],
-        type_configuration_model_cls: Optional[BaseModel],
+        type_configuration_model_cls: Optional[Type[BaseModel]],
     ) -> BaseResourceHandlerRequest:
         # pylint: disable=protected-access
         return BaseResourceHandlerRequest(