type

openvinotoolkit · Dec 24, 2024 · cba3aba · cba3aba
1 parent 79ac83b
commit cba3aba
Show file tree

Hide file tree

Showing 2 changed files with 4 additions and 4 deletions.
diff --git a/nncf/common/graph/graph.py b/nncf/common/graph/graph.py
@@ -82,7 +82,7 @@ def node_name(self) -> NNCFNodeName:
         return cast(NNCFNodeName, self._attributes[NNCFNode.NODE_NAME_ATTR])
 
     @property
-    def metatype(self) -> OperatorMetatype:
+    def metatype(self) -> Type[OperatorMetatype]:
         return cast(OperatorMetatype, self._attributes[NNCFNode.METATYPE_ATTR])
 
     @property

diff --git a/nncf/common/graph/utils.py b/nncf/common/graph/utils.py
@@ -10,7 +10,7 @@
 # limitations under the License.
 
 from functools import partial
-from typing import List, Set, Tuple, Union
+from typing import List, Set, Tuple, Type, Union
 
 from nncf.common.graph import NNCFGraph
 from nncf.common.graph import NNCFNode
@@ -90,8 +90,8 @@ def get_split_axis(input_shapes: List[List[int]], output_shapes: List[List[int]]
 
 def get_number_of_quantized_ops(
     graph: NNCFGraph,
-    quantizer_metatypes: List[OperatorMetatype],
-    quantizable_metatypes: List[OperatorMetatype],
+    quantizer_metatypes: List[Type[OperatorMetatype]],
+    quantizable_metatypes: List[Type[OperatorMetatype]],
 ) -> int:
     """
     Returns the number of quantized operations in the graph.