@@ -287,10 +287,10 @@ def messages_to_module(
287287 use_isolating : bool = True ,
288288 functions : Mapping [str , Callable ] | None = None ,
289289 escapers : Sequence [IsEscaper ] | None = None ,
290- ) -> tuple :
290+ ) -> tuple [ codegen . Module , dict [ str , str ], dict [ str , object ], list [ CompilationErrorItem ]] :
291291 """
292292 Compile a set of {id: Message/Term objects} to a Python module, returning a tuple:
293- (codegen.Module object, dictionary mapping message IDs to Python functions ,
293+ (codegen.Module object, dictionary mapping message IDs to Python function names ,
294294 module globals dictionary, errors list)
295295 """
296296 if functions is None :
@@ -405,7 +405,7 @@ def get_name_properties(name: str) -> dict[str, object]:
405405 function = compile_message (msg , msg_id , function_name , module , compiler_env )
406406 module .add_function (function_name , function )
407407
408- module = codegen .simplify (module , Simplifier (compiler_env ))
408+ module = checked_cast ( codegen .Module , codegen . simplify (module , Simplifier (compiler_env ) ))
409409 return (module , compiler_env .message_mapping , module_globals , compiler_env .errors )
410410
411411
0 commit comments