replace schemaVx with current schema

yarden-sony · yarden-sony · commit 4099b0badb74 · 2025-04-01T09:46:18.000+03:00
diff --git a/tests/common_tests/helpers/tpcs_for_tests/v4/tpc.py b/tests/common_tests/helpers/tpcs_for_tests/v4/tpc.py
@@ -14,8 +14,7 @@
 # ==============================================================================
 from typing import List, Tuple
 
-import model_compression_toolkit as mct
-import model_compression_toolkit.target_platform_capabilities.schema.v1 as schema
+import model_compression_toolkit.target_platform_capabilities.schema.mct_current_schema as schema
 from mct_quantizers import QuantizationMethod
 from model_compression_toolkit.constants import FLOAT_BITWIDTH
 from model_compression_toolkit.target_platform_capabilities.constants import KERNEL_ATTR, BIAS_ATTR, WEIGHTS_N_BITS, \
diff --git a/tests_pytest/common_tests/unit_tests/core/quantization/quantization_params_selection/test_qarams_activations_computation.py b/tests_pytest/common_tests/unit_tests/core/quantization/quantization_params_selection/test_qarams_activations_computation.py
@@ -33,7 +33,7 @@
     get_activations_qparams
 )
 from model_compression_toolkit.target_platform_capabilities import Signedness, OpQuantizationConfig
-from model_compression_toolkit.target_platform_capabilities.schema.v1 import AttributeQuantizationConfig
+from model_compression_toolkit.target_platform_capabilities.schema.mct_current_schema import AttributeQuantizationConfig
 
 
 class TestActivationQParams:

Original file line number	Diff line number	Diff line change
`@@ -33,7 +33,7 @@`
`33`	`33`	`get_activations_qparams`
`34`	`34`	`)`
`35`	`35`	`from model_compression_toolkit.target_platform_capabilities import Signedness, OpQuantizationConfig`
`36`		`-from model_compression_toolkit.target_platform_capabilities.schema.v1 import AttributeQuantizationConfig`
	`36`	`+from model_compression_toolkit.target_platform_capabilities.schema.mct_current_schema import AttributeQuantizationConfig`
`37`	`37`
`38`	`38`
`39`	`39`	`class TestActivationQParams:`