Skip to content

Commit 26ec4fe

Browse files
authored
Merge pull request #133 from adf-python/fix/type-hint
Add type hints
2 parents c902805 + 6b8af6b commit 26ec4fe

1 file changed

Lines changed: 8 additions & 6 deletions

File tree

src/adf_core_python/core/component/module/abstract_module.py

Lines changed: 8 additions & 6 deletions
Original file line numberDiff line numberDiff line change
@@ -4,6 +4,8 @@
44
from abc import ABC, abstractmethod
55
from typing import TYPE_CHECKING
66

7+
from structlog.stdlib import BoundLogger
8+
79
from adf_core_python.core.logger.logger import get_agent_logger
810

911
if 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

Comments
 (0)