diff options
Diffstat (limited to 'src/mlia/core/common.py')
-rw-r--r-- | src/mlia/core/common.py | 4 |
1 files changed, 3 insertions, 1 deletions
diff --git a/src/mlia/core/common.py b/src/mlia/core/common.py index a11bf9a..63fb324 100644 --- a/src/mlia/core/common.py +++ b/src/mlia/core/common.py @@ -5,6 +5,8 @@ This module contains common interfaces/classess shared across core module. """ +from __future__ import annotations + from abc import ABC from abc import abstractmethod from enum import auto @@ -30,7 +32,7 @@ class AdviceCategory(Flag): ALL = OPERATORS | PERFORMANCE | OPTIMIZATION @classmethod - def from_string(cls, value: str) -> "AdviceCategory": + def from_string(cls, value: str) -> AdviceCategory: """Resolve enum value from string value.""" category_names = [item.name for item in AdviceCategory] if not value or value.upper() not in category_names: |