diff options
author | Dmitrii Agibov <dmitrii.agibov@arm.com> | 2022-09-08 14:24:39 +0100 |
---|---|---|
committer | Dmitrii Agibov <dmitrii.agibov@arm.com> | 2022-09-09 17:21:48 +0100 |
commit | f5b293d0927506c2a979a091bf0d07ecc78fa181 (patch) | |
tree | 4de585b7cb6ed34da8237063752270189a730a41 /src/mlia/core/context.py | |
parent | cde0c6ee140bd108849bff40467d8f18ffc332ef (diff) | |
download | mlia-f5b293d0927506c2a979a091bf0d07ecc78fa181.tar.gz |
MLIA-386 Simplify typing in the source code
- Enable deferred annotations evaluation
- Use builtin types for type hints whenever possible
- Use | syntax for union types
- Rename mlia.core._typing into mlia.core.typing
Change-Id: I3f6ffc02fa069c589bdd9e8bddbccd504285427a
Diffstat (limited to 'src/mlia/core/context.py')
-rw-r--r-- | src/mlia/core/context.py | 31 |
1 files changed, 15 insertions, 16 deletions
diff --git a/src/mlia/core/context.py b/src/mlia/core/context.py index 83d2f7c..a4737bb 100644 --- a/src/mlia/core/context.py +++ b/src/mlia/core/context.py @@ -7,15 +7,14 @@ Context is an object that describes advisor working environment and requested behavior (advice categories, input configuration parameters). """ +from __future__ import annotations + import logging from abc import ABC from abc import abstractmethod from pathlib import Path from typing import Any -from typing import List from typing import Mapping -from typing import Optional -from typing import Union from mlia.core.common import AdviceCategory from mlia.core.events import DefaultEventPublisher @@ -50,7 +49,7 @@ class Context(ABC): @property @abstractmethod - def event_handlers(self) -> Optional[List[EventHandler]]: + def event_handlers(self) -> list[EventHandler] | None: """Return list of the event_handlers.""" @property @@ -60,7 +59,7 @@ class Context(ABC): @property @abstractmethod - def config_parameters(self) -> Optional[Mapping[str, Any]]: + def config_parameters(self) -> Mapping[str, Any] | None: """Return configuration parameters.""" @property @@ -73,7 +72,7 @@ class Context(ABC): self, *, advice_category: AdviceCategory, - event_handlers: List[EventHandler], + event_handlers: list[EventHandler], config_parameters: Mapping[str, Any], ) -> None: """Update context parameters.""" @@ -98,14 +97,14 @@ class ExecutionContext(Context): self, *, advice_category: AdviceCategory = AdviceCategory.ALL, - config_parameters: Optional[Mapping[str, Any]] = None, - working_dir: Optional[Union[str, Path]] = None, - event_handlers: Optional[List[EventHandler]] = None, - event_publisher: Optional[EventPublisher] = None, + config_parameters: Mapping[str, Any] | None = None, + working_dir: str | Path | None = None, + event_handlers: list[EventHandler] | None = None, + event_publisher: EventPublisher | None = None, verbose: bool = False, logs_dir: str = "logs", models_dir: str = "models", - action_resolver: Optional[ActionResolver] = None, + action_resolver: ActionResolver | None = None, ) -> None: """Init execution context. @@ -151,22 +150,22 @@ class ExecutionContext(Context): self._advice_category = advice_category @property - def config_parameters(self) -> Optional[Mapping[str, Any]]: + def config_parameters(self) -> Mapping[str, Any] | None: """Return configuration parameters.""" return self._config_parameters @config_parameters.setter - def config_parameters(self, config_parameters: Optional[Mapping[str, Any]]) -> None: + def config_parameters(self, config_parameters: Mapping[str, Any] | None) -> None: """Setter for the configuration parameters.""" self._config_parameters = config_parameters @property - def event_handlers(self) -> Optional[List[EventHandler]]: + def event_handlers(self) -> list[EventHandler] | None: """Return list of the event handlers.""" return self._event_handlers @event_handlers.setter - def event_handlers(self, event_handlers: List[EventHandler]) -> None: + def event_handlers(self, event_handlers: list[EventHandler]) -> None: """Setter for the event handlers.""" self._event_handlers = event_handlers @@ -196,7 +195,7 @@ class ExecutionContext(Context): self, *, advice_category: AdviceCategory, - event_handlers: List[EventHandler], + event_handlers: list[EventHandler], config_parameters: Mapping[str, Any], ) -> None: """Update context parameters.""" |