get_applied_transformation_layout -> transformation_layout

openvinotoolkit · Apr 22, 2024 · a6abd0a · a6abd0a
1 parent 9e62cbf
commit a6abd0a
Show file tree

Hide file tree

Showing 2 changed files with 5 additions and 5 deletions.
diff --git a/nncf/torch/nncf_network.py b/nncf/torch/nncf_network.py
@@ -784,7 +784,7 @@ def _collect_eval_op_scopes(self, model: nn.Module, dummy_forward_fn: Callable)
                 result.append(scope_in_model)
         return result
 
-    def get_applied_transformation_layout(self) -> PTTransformationLayout:
+    def transformation_layout(self) -> PTTransformationLayout:
         """
         Collects all hooks applied to the NNCFNetwork, converts them to insertion commands
         and returns in PTTransformationLayout format. Default hooks group name is used in

diff --git a/tests/torch/nncf_network/test_get_applied_modifications.py b/tests/torch/nncf_network/test_get_applied_modifications.py
@@ -100,7 +100,7 @@ def test_get_applied_modification_commands(
     layout.register(command)
     model_transformer.transform(layout)
 
-    applied_commands = nncf_model.nncf.get_applied_transformation_layout()
+    applied_commands = nncf_model.nncf.transformation_layout()
 
     assert len(applied_commands.transformations) == 1
     applied_command = applied_commands.transformations[0]
@@ -130,7 +130,7 @@ def test_priority_of_get_applied_modification_commands(
 
     model_transformer.transform(layout)
 
-    applied_commands = nncf_model.nncf.get_applied_transformation_layout()
+    applied_commands = nncf_model.nncf.transformation_layout()
     assert len(applied_commands.transformations) == len(commands)
     for applied_command in applied_commands.transformations:
         command = commands[applied_command.priority]
@@ -153,7 +153,7 @@ def test_all_possible_combinations_of_commands_for_get_applied_commands(
 
     model_transformer.transform(commands)
 
-    applied_commands = nncf_model.nncf.get_applied_transformation_layout()
+    applied_commands = nncf_model.nncf.transformation_layout()
     assert len(applied_commands.transformations) == len(commands.transformations)
     for command in commands.transformations:
         _translate_target_types(trace_parameters, command)
@@ -186,4 +186,4 @@ def test_get_applied_modification_commands_broken_call_hook(
 
     nncf_model.nncf.external_op.clear()
     with pytest.raises(AssertionError):
-        nncf_model.nncf.get_applied_transformation_layout()
+        nncf_model.nncf.transformation_layout()