diff options
Diffstat (limited to 'src/mlia/core/context.py')
| -rw-r--r-- | src/mlia/core/context.py | 6 |
1 files changed, 3 insertions, 3 deletions
diff --git a/src/mlia/core/context.py b/src/mlia/core/context.py index 8b3dd2c..83d2f7c 100644 --- a/src/mlia/core/context.py +++ b/src/mlia/core/context.py @@ -55,7 +55,7 @@ class Context(ABC): @property @abstractmethod - def advice_category(self) -> Optional[AdviceCategory]: + def advice_category(self) -> AdviceCategory: """Return advice category.""" @property @@ -97,7 +97,7 @@ class ExecutionContext(Context): def __init__( self, *, - advice_category: Optional[AdviceCategory] = None, + 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, @@ -141,7 +141,7 @@ class ExecutionContext(Context): self._action_resolver = action_resolver or APIActionResolver() @property - def advice_category(self) -> Optional[AdviceCategory]: + def advice_category(self) -> AdviceCategory: """Return advice category.""" return self._advice_category |