From b1cc0d1299051068dd6e3d08d28ff95c33a504a2 Mon Sep 17 00:00:00 2001 From: liord Date: Sun, 12 Jan 2025 15:04:36 +0200 Subject: [PATCH] Fix test --- tests/pytorch_tests/model_tests/feature_models/matmul_test.py | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/tests/pytorch_tests/model_tests/feature_models/matmul_test.py b/tests/pytorch_tests/model_tests/feature_models/matmul_test.py index c457a9319..f2fe4ce74 100644 --- a/tests/pytorch_tests/model_tests/feature_models/matmul_test.py +++ b/tests/pytorch_tests/model_tests/feature_models/matmul_test.py @@ -15,7 +15,7 @@ import torch from tests.pytorch_tests.model_tests.base_pytorch_test import BasePytorchTest from model_compression_toolkit.target_platform_capabilities.tpc_models.imx500_tpc.latest import generate_pytorch_tpc -from tests.common_tests.helpers.generate_test_tp_model import generate_test_tp_model +from tests.common_tests.helpers.generate_test_tpc import generate_test_tpc """ This test checks the MatMul substitution function. @@ -60,7 +60,7 @@ def get_tpc(self): return { 'no_quantization': generate_pytorch_tpc( name="no_quant_pytorch_test", - tp_model=generate_test_tp_model( + tpc=generate_test_tpc( { 'weights_n_bits': 32, 'activation_n_bits': 32,