44from abc import ABC , abstractmethod
55from typing import TYPE_CHECKING
66
7+ from structlog .stdlib import BoundLogger
8+
79from adf_core_python .core .logger .logger import get_agent_logger
810
911if TYPE_CHECKING :
@@ -25,17 +27,17 @@ def __init__(
2527 module_manager : ModuleManager ,
2628 develop_data : DevelopData ,
2729 ) -> None :
28- self ._agent_info = agent_info
29- self ._world_info = world_info
30- self ._scenario_info = scenario_info
31- self ._module_manager = module_manager
32- self ._develop_data = develop_data
30+ self ._agent_info : AgentInfo = agent_info
31+ self ._world_info : WorldInfo = world_info
32+ self ._scenario_info : ScenarioInfo = scenario_info
33+ self ._module_manager : ModuleManager = module_manager
34+ self ._develop_data : DevelopData = develop_data
3335 self ._count_precompute : int = 0
3436 self ._count_resume : int = 0
3537 self ._count_prepare : int = 0
3638 self ._count_update_info : int = 0
3739 self ._count_update_info_current_time : int = 0
38- self ._logger = get_agent_logger (
40+ self ._logger : BoundLogger = get_agent_logger (
3941 f"{ self .__class__ .__module__ } .{ self .__class__ .__qualname__ } " ,
4042 self ._agent_info ,
4143 )
0 commit comments