Added Type[BaseModel] Annotation

Author: ammokhov

python/rpdk/python/codegen.py

@@ -170,7 +170,7 @@ def generate(self, project):
         LOG.debug("Generate complete")
 
     def _pre_package(self, build_path):
-        f = TemporaryFile("w+b")  # pylint: disable=R1732
+        f = TemporaryFile("w+b")
 
         with zipfile.ZipFile(f, mode="w") as zip_file:
             self._recursive_relative_write(build_path, build_path, zip_file)

src/cloudformation_cli_python_lib/resource.py

@@ -65,7 +65,7 @@ def __init__(
         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(