Comment

openvinotoolkit · Nov 14, 2024 · 9e946ee · 9e946ee
1 parent 04290c2
commit 9e946ee
Show file tree

Hide file tree

Showing 2 changed files with 4 additions and 7 deletions.
diff --git a/tests/torch/conftest.py b/tests/torch/conftest.py
@@ -124,13 +124,9 @@ def pytest_addoption(parser: Parser):
 
 
 def pytest_configure(config: Config):
-    regen_dot = config.getoption("--regen-dot", False)
-    if regen_dot:
-        os.environ["NNCF_TEST_REGEN_DOT"] = "1"
-
-    regen_json = config.getoption("--regen-json", False)
-    if regen_json:
-        os.environ["NNCF_TEST_REGEN_JSON"] = "1"
+    for regen_option in ["dot", "json"]:
+        if config.getoption(f"--regen-{regen_option}", False):
+            os.environ[f"NNCF_TEST_REGEN_{regen_option.upper()}"] = "1"
 
 
 @pytest.fixture(scope="module")

diff --git a/tests/torch/fx/test_bias_correction.py b/tests/torch/fx/test_bias_correction.py
@@ -217,4 +217,5 @@ def test__get_subgraph_data_for_node(self, quantized_test_model, layer_name, ref
         ),
     )
     def test_verify_collected_stat_inputs_map(self, model_cls, ref_stat_inputs_map, tmpdir):
+        breakpoint()
         return super().test_verify_collected_stat_inputs_map(model_cls, ref_stat_inputs_map, tmpdir)