Comments

Author: daniil-lyakhov

src/nncf/parameters.py

@@ -10,6 +10,7 @@
 # limitations under the License.
 
 from enum import Enum
+from typing import Any
 
 from nncf.common.utils.api_marker import api
 
@@ -18,6 +19,10 @@ class StrEnum(str, Enum):
     def __str__(self) -> str:
         return str(self.value)
 
+    @staticmethod
+    def _generate_next_value_(name: str, start: int, count: int, last_values: list[Any]) -> Any:
+        return name.lower()
+
 
 @api(canonical_alias="nncf.TargetDevice")
 class TargetDevice(StrEnum):

src/nncf/tensor/definitions.py

@@ -12,7 +12,7 @@
 from dataclasses import dataclass
 from enum import Enum
 from enum import auto
-from typing import Any, Optional, Union
+from typing import Optional, Union
 
 from nncf.parameters import StrEnum
 
@@ -38,10 +38,6 @@ class TensorDataType(StrEnum):
     Enum representing the different tensor data types.
     """
 
-    @staticmethod
-    def _generate_next_value_(name: str, start: int, count: int, last_values: list[Any]) -> Any:
-        return name.lower()
-
     float16 = auto()
     bfloat16 = auto()
     float32 = auto()