Fix exception type

openvinotoolkit · Feb 14, 2024 · 37ac299 · 37ac299
1 parent 9957b5c
commit 37ac299
Showing 1 changed file with 2 additions and 1 deletion.
diff --git a/tests/common/quantization/test_quantizer_propagation_solver.py b/tests/common/quantization/test_quantizer_propagation_solver.py
@@ -18,6 +18,7 @@
 import networkx as nx
 import pytest
 
+import nncf
 from nncf.common.graph import NNCFGraph
 from nncf.common.graph import NNCFNode
 from nncf.common.graph import NNCFNodeName
@@ -2116,7 +2117,7 @@ def get_metatype_vs_quantizer_configs_map(
 
         register_us_group_w_t_spy = mocker.spy(MultiConfigQuantizerSetup, "register_unified_scale_group_with_types")
         if test_struct.ref_qp_vs_unified_scale_type is None:
-            with pytest.raises(RuntimeError):
+            with pytest.raises(nncf.InternalError):
                 quant_proposal = quant_prop_solver.run_on_ip_graph(ip_graph)
         else:
             quant_proposal = quant_prop_solver.run_on_ip_graph(ip_graph)