Add more API entities

nncf/data/dataset.py

@@ -18,11 +18,13 @@
 from typing import Generic
 from typing import TypeVar
 
+from nncf.common.api_marker import api
 
 DataItem = TypeVar('DataItem')
 ModelInput = TypeVar('ModelInput')
 
 
+@api()
 class Dataset(Generic[DataItem, ModelInput]):
     """
     The `nncf.Dataset` class defines the interface by which compression algorithms

nncf/parameters.py

@@ -13,6 +13,10 @@
 
 from enum import Enum
 
+from nncf.common.api_marker import api
+
+
+@api()
 class TargetDevice(Enum):
     """
     Describes the target device the specificity of which will be taken
@@ -32,6 +36,7 @@ class TargetDevice(Enum):
     CPU_SPR = 'CPU_SPR'
 
 
+@api()
 class ModelType(Enum):
     """
     Describes the model type the specificity of which will be taken into

nncf/scopes.py

@@ -14,10 +14,12 @@
 import re
 from typing import List, Optional
 
+from nncf.common.api_marker import api
 from nncf.common.logging import nncf_logger
 from nncf.common.graph.graph import NNCFGraph
 
 
+@api()
 class IgnoredScope:
     """
     Dataclass that contains description of the ignored scope.
@@ -114,11 +116,11 @@ def get_ignored_node_names_from_ignored_scope(ignored_scope: IgnoredScope,
             if ignored_node_name in node_names:
                 matched_by_names.append(ignored_node_name)
         if strict and len(ignored_scope.names) != len(matched_by_names):
-                skipped_names = set(ignored_scope.names) - set(matched_by_names)
-                raise RuntimeError(f'Ignored nodes with name {list(skipped_names)}'
-                                    ' were not found in the NNCFGraph. ' + error_msg)
+            skipped_names = set(ignored_scope.names) - set(matched_by_names)
+            raise RuntimeError(f'Ignored nodes with name {list(skipped_names)}'
+                               ' were not found in the NNCFGraph. ' + error_msg)
         nncf_logger.info(f'{len(matched_by_names)}'
-                          ' ignored nodes was found by name in the NNCFGraph')
+                         ' ignored nodes was found by name in the NNCFGraph')
 
     matched_by_patterns = []
     if ignored_scope.patterns: