@@ -287,10 +287,10 @@ def messages_to_module(
287
287
use_isolating : bool = True ,
288
288
functions : Mapping [str , Callable ] | None = None ,
289
289
escapers : Sequence [IsEscaper ] | None = None ,
290
- ) -> tuple :
290
+ ) -> tuple [ codegen . Module , dict [ str , str ], dict [ str , object ], list [ CompilationErrorItem ]] :
291
291
"""
292
292
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 ,
294
294
module globals dictionary, errors list)
295
295
"""
296
296
if functions is None :
@@ -405,7 +405,7 @@ def get_name_properties(name: str) -> dict[str, object]:
405
405
function = compile_message (msg , msg_id , function_name , module , compiler_env )
406
406
module .add_function (function_name , function )
407
407
408
- module = codegen .simplify (module , Simplifier (compiler_env ))
408
+ module = checked_cast ( codegen .Module , codegen . simplify (module , Simplifier (compiler_env ) ))
409
409
return (module , compiler_env .message_mapping , module_globals , compiler_env .errors )
410
410
411
411
0 commit comments