From a1831422e1af531a8855a3caa60513be9eee62aa Mon Sep 17 00:00:00 2001
From: Vasily Shamporov <vasily.shamporov@intel.com>
Date: Thu, 13 Apr 2023 16:39:18 +0200
Subject: [PATCH] API change

---
 nncf/torch/model_creation.py | 3 ++-
 1 file changed, 2 insertions(+), 1 deletion(-)

diff --git a/nncf/torch/model_creation.py b/nncf/torch/model_creation.py
index 275de53dc65..88882d5cae0 100644
--- a/nncf/torch/model_creation.py
+++ b/nncf/torch/model_creation.py
@@ -53,7 +53,8 @@ def create_compressed_model(model: Module,
                             dummy_forward_fn: Callable[[Module], Any] = None,
                             wrap_inputs_fn: Callable[[Tuple, Dict], Tuple[Tuple, Dict]] = None,
                             wrap_outputs_fn: Callable[[Tuple, Dict], Tuple[Tuple, Dict]] = None,
-                            dump_graphs=True) \
+                            dump_graphs=True,
+                            new_arg = False) \
     -> Tuple[CompressionAlgorithmController, NNCFNetwork]:
     """
     The main function used to produce a model ready for compression fine-tuning from an original PyTorch