fix failed tests

sony · Mar 20, 2024 · c26e157 · c26e157
1 parent d011e5e
commit c26e157
Show file tree

Hide file tree

Showing 2 changed files with 2 additions and 2 deletions.
diff --git a/tests/keras_tests/non_parallel_tests/test_lp_search_bitwidth.py b/tests/keras_tests/non_parallel_tests/test_lp_search_bitwidth.py
@@ -71,7 +71,7 @@ def __init__(self, layer_to_ru_mapping):
         self.config_reconstruction_helper = MockReconstructionHelper()
         self.non_conf_ru_dict = None
 
-    def compute_ru_matrix(self, target):
+    def compute_resource_utilization_matrix(self, target):
         # minus 1 is normalization by the minimal resource utilization (which is always 1 in this test)
         if target == RUTarget.WEIGHTS:
             ru_matrix = [np.flip(np.array([ru.weights_memory - 1 for _, ru in self.layer_to_ru_mapping[0].items()]))]

diff --git a/tests/pytorch_tests/function_tests/test_doc_examples.py b/tests/pytorch_tests/function_tests/test_doc_examples.py
@@ -32,4 +32,4 @@ def test_pytorch_gptq_facade(self):
         doctest.testfile("quantization_facade.py", package=gptq.pytorch, verbose=True, raise_on_error=RAISE_ON_ERROR)
 
     def test_pytorch_resource_utilization_data_facade(self):
-        doctest.testfile("resource_utilization_data_test.py", package=core.pytorch, verbose=True, raise_on_error=RAISE_ON_ERROR)
+        doctest.testfile("resource_utilization_data.py", package=core.pytorch, verbose=True, raise_on_error=RAISE_ON_ERROR)